首页 > 技术文章 > AI换脸技术源码

liyanyan665 2019-08-08 18:17 原文

import requests
import simplejson
import json
import base64

def find_face(imgpath):
print("finding")

http_url = 'https://api-cn.faceplusplus.com/facepp/v3/detect'

data = {"api_key": "https://console.faceplusplus.com.cn申请'',
"api_secret": 'https://console.faceplusplus.com.cn申请', "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']

# print(rectangle)

return rectangle


# number表示换脸的相似度
def merge_face(image_url_1, image_url_2, image_url, number):
ff1 = find_face(image_url_1)

ff2 = find_face(image_url_2)

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_url_1, 'rb')

f1_64 = base64.b64encode(f1.read())

f1.close()

f2 = open(image_url_2, 'rb')

f2_64 = base64.b64encode(f2.read())

f2.close()

data = {"api_key": 'https://console.faceplusplus.com.cn申请L', "api_secret": 'https://console.faceplusplus.com.cn申请',
"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_con = response.content.decode('utf-8')

req_dict = json.JSONDecoder(http://www.my516.com).decode(req_con)

result = req_dict['result']

imgdata = base64.b64decode(result)

file = open(image_url, 'wb')

file.write(imgdata)

file.close()

#E:\test\test
image1 = r"E:\test\test\cover1.jpg"

image2 = r"E:\test\test\cover11.jpg"

image = r"E:\test\test\face7.jpg"

merge_face(image2, image1, image, 100)
---------------------

推荐阅读