Python中怎么實現(xiàn)一個換臉功能

功能實現(xiàn)

實現(xiàn)換臉功能，我們大致可以分為兩種：一種是所有功能都通過自己編碼來實現(xiàn)，另一種是借助于第三方 API 來實現(xiàn)，第一種方式可能需要我們進(jìn)行大量的編碼才能實現(xiàn)，而第二種方式我們只需進(jìn)行少量的編碼即可實現(xiàn)。

本文我們使用更簡單的第二種方式來實現(xiàn)，我們用到的 API 接口提供方是 Face++，首先我們需要到該網(wǎng)站注冊一個自己的賬號，注冊地址為：https://console.faceplusplus.com.cn/register，打開后如下所示：

我們可以通過手機號和郵箱兩種方式來注冊，注冊好賬號之后，我們再到登錄地址 https://console.faceplusplus.com.cn/login 進(jìn)行登錄，登錄之后，我們會發(fā)現(xiàn)網(wǎng)站已經(jīng)為我們創(chuàng)建好了應(yīng)用，如下圖所示：

我們需要用到的是上圖中的 API Key 和 API Secret 的值，下面來看一下具體實現(xiàn)代碼：

# 獲取人臉關(guān)鍵點
def find_face(imgpath):
    print("正在查找……")
    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)
    print(this_json2)
    faces = this_json2['faces']
    list0 = faces[0]
    rectangle = list0['face_rectangle']
    # print(rectangle)
    return rectangle


# 換臉，圖片的大小應(yīng)不超過 2M，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'])
    print(rectangle2)
    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()