python_face_changing/face.py at master · lvchuandong/python_face_changing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import requests
import simplejson
import json
import base64


def find_face(imgpath):

    print("finding")
    #调用相应的api
    http_url = 'https://api-cn.faceplusplus.com/facepp/v3/detect'
    #获取自己的账号
    data = {"api_key": 'rnoaJExcaeDt5H3_QgTfS61ZWEsWRW1B',
        "api_secret": 'Vkv5lSzXgM5syr0guXuDYTshzQQRSM9f', "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


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": 'rnoaJExcaeDt5H3_QgTfS61ZWEsWRW1B', "api_secret": 'Vkv5lSzXgM5syr0guXuDYTshzQQRSM9f',
        "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().decode(req_con)

    result = req_dict['result']

    imgdata = base64.b64decode(result)

    file = open(image_url, 'wb')

    file.write(imgdata)

    file.close()


def test():

    image1 = "2.jpg"

    image2 = "1.jpg"

    image =  "output.jpg"

    merge_face(image2,image1,image,50)


if __name__ == '__main__':
	test()