不到 100 行 Python 代码徐峥变葛优

Posted by Python小二 on 2020-04-29

给照片换脸大家应该都见过,本文我们来介绍一下如何通过 Python 实现换脸。

1. 功能实现

实现换脸功能,我们大致可以分为两种:一种是所有功能都通过自己编码来实现,另一种是借助于第三方 API 来实现,第一种方式可能需要我们进行大量的编码才能实现,而第二种方式我们只需进行少量的编码即可实现。

本文我们使用更简单的第二种方式来实现,我们用到的 API 接口提供方是 Face++,首先我们需要到该网站注册一个自己的账号,注册地址为:https://console.faceplusplus.com.cn/register,打开后如下所示:

我们可以通过手机号和邮箱两种方式来注册,注册好账号之后,我们再到登录地址 https://console.faceplusplus.com.cn/login 进行登录,登录之后,我们会发现网站已经为我们创建好了应用,如下图所示:

我们需要用到的是上图中的 API KeyAPI Secret 的值,下面来看一下具体实现代码:

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
import requests, simplejson, json, base64

# 获取人脸关键点
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


# 换脸,图片的大小应不超过 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()

2. 效果展示

下面我们通过具体图片来看一下实现效果。

首先,我们使用两位男明星的图片进行效果展示,以徐峥和葛优为例,原图如下所示:

换脸后的效果图如下所示:

接着,我们再使用两位女明星的图片进行效果展示,以贾玲和关晓彤为例,原图如下所示:

换脸后的效果图如下所示:

是不是感觉有点意思,有兴趣的同学可以自己动手试试。

欢迎微信搜索 Python小二,第一时间阅读、获取源码,回复关键字 1024 可以免费领取个人整理的各类编程语言学习资料。