提交代码

Author: cxhuan

xianhuan/changeFace/changeface.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+@author: 闲欢
+"""
+import base64
+import requests
+import json
+import simplejson
+
+# 第一步，获取人脸关键点
+api_key = '你的key'
+api_secret = '你的secret'
+
+def find_face(imgpath):
+    http_url = 'https://api-cn.faceplusplus.com/facepp/v3/detect'
+    data = {"api_key": api_key,
+            "api_secret": api_secret,
+            "image_url": imgpath,
+            "return_landmark": 1  # 是否检测并返回人脸关键点
+            }
+    files = {"image_file": open(imgpath, "rb")}
+    response = requests.post(http_url, data=data, files=files)
+    req_con = response.content.decode('utf-8')
+    req_dict = json.JSONDecoder().decode(req_con)
+    this_json = simplejson.dumps(req_dict)
+    this_json2 = simplejson.loads(this_json)
+    faces = this_json2['faces']
+    list0 = faces[0]
+    rectangle = list0['face_rectangle']
+    return rectangle
+
+# 第二步，换脸
+# number表示换脸的相似度
+def merge_face(image_url1, image_url2, image_url, number):
+    ff1 = find_face(image_url1)
+    ff2 = find_face(image_url2)
+
+    rectangle1 = str(str(ff1['top']) + "," + str(ff1['left']) + "," + str(ff1['width']) + "," + str(ff1['height']))
+    rectangle2 = str(ff2['top']) + "," + str(ff2['left']) + "," + str(ff2['width']) + "," + str(ff2['height'])
+
+    url_add = "https://api-cn.faceplusplus.com/imagepp/v1/mergeface"
+    f1 = open(image_url1, 'rb')
+    f1_64 = base64.b64encode(f1.read())
+    f1.close()
+    f2 = open(image_url2, 'rb')
+    f2_64 = base64.b64encode(f2.read())
+    f2.close()
+
+    data = {"api_key": api_key,
+            "api_secret": api_secret,
+            "template_base64": f1_64,
+            "template_rectangle": rectangle1,
+            "merge_base64": f2_64,
+            "merge_rectangle": rectangle2,
+            "merge_rate": number
+            }
+    response = requests.post(url_add, data=data)
+    req_con1 = response.content.decode('utf-8')
+    req_dict = json.JSONDecoder().decode(req_con1)
+    result = req_dict['result']
+    imgdata = base64.b64decode(result)
+    file = open(image_url, 'wb')
+    file.write(imgdata)
+    file.close()
+
+image1 = r"C:\Users\xx\Downloads\tmp\pic/2.jpg"
+image2 = r"C:\Users\xx\Downloads\tmp\pic/3.jpg"
+image = r"C:\Users\xx\Downloads\tmp\pic/5.jpg"
+
+merge_face(image1, image2, image, 90)