使用 python 進行 面部合成

 

 完整代碼已上傳至GitHub:

https://github.com/chestnut-egg/Face

 

一. 準備工做

1. 此程序使用的是 Face++ 的API,因此須要去Face++官網註冊帳號:

https://www.faceplusplus.com.cn/

2. 建立應用,獲取 key 和 secret

3. 下載 simplejson 模塊 ,使用pip就能夠下載了

pip install simplejson

 

二. 程序思路

 1. 使用 decect 接口,獲取人臉關鍵點

接口詳細文檔:

https://console.faceplusplus.com.cn/documents/4888373

* return_landmark 參數 不能爲 0 否則不會返回人臉關鍵點

return_landmark

Int

是否檢測並返回人臉關鍵點。合法值爲： 2  檢測。返回 106 我的臉關鍵點。 1 檢測。返回 83 我的臉關鍵點。 0 不檢測 注：本參數默認值爲 0

核心代碼:

def find_face(imgpath):
    print("finding")
    http_url = 'https://api-cn.faceplusplus.com/facepp/v3/detect'
    data = {"api_key": key, "api_secret": 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 = 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

 

2. 使用 mergeface 接口,合成臉部圖像

接口詳細文檔:

https://console.faceplusplus.com.cn/documents/20813963

* 注意圖片文件大小不超過 2 MB

核心代碼:

# 模板圖片地址 合成圖片地址 生成圖片地址 合成指數0-100
def add_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'])

    # print(rectangle1)
    # print(rectangle2)

    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": key, "api_secret": 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_con = response.content.decode('utf-8')
    req_dict = JSONDecoder().decode(req_con)
    print(req_dict)
    result = req_dict['result']
    imgdata = base64.b64decode(result)
    file = open(image_url, 'wb')
    file.write(imgdata)
    file.close()

 

3. 示例運行代碼

# 單獨兩張照片的合成示例

image_url_1 = r"C:\Users\1.jpg"
image_url_2 = r"C:\Users\2.jpg"
image_url = r'C:\Users\result.jpg'
add_face(image_url_1,image_url_2,image_url,50)

 

4. 封裝一個多張照片的合成函數

用 列表List 儲存圖片地址,先以最開始的兩張進行合成,而後將合成後的圖片與列表中的其餘圖像依次合成

* 程序沒有作List的長度驗證,注意邊界特殊狀況

def add_many(list_face):

    print("正在合成第1-2張")
    image_now = r'C:\Users\now.jpg'
    add_face(list_face[0], list_face[1], image_now, 50)

    for index in range(2,len(list_face)):
        print("正在合成第"+str(index+1)+"張")
        add_face(image_now, list_face[index], image_now, 50)

 

5. 成果展現

素材1:

素材2:

合成結果:

 

完整代碼:

https://github.com/chestnut-egg/Face