前言 和換臉演算法的簡單粗暴相比,人臉融合演算法就要彈性的多。顧名思義,人臉融合是對兩張人臉進行融合處理,生成的人臉同時具 備兩張人臉的外貌特征。人臉融合有什麼實際意義呢?一個簡單的應用就是用父母雙方的臉部圖片融合,得到未來孩子可能的長相。 本文通過百度AI開放平臺的人臉融合功能來做一個簡單的試驗。 前 ...
前言
和換臉演算法的簡單粗暴相比,人臉融合演算法就要彈性的多。顧名思義,人臉融合是對兩張人臉進行融合處理,生成的人臉同時具
備兩張人臉的外貌特征。人臉融合有什麼實際意義呢?一個簡單的應用就是用父母雙方的臉部圖片融合,得到未來孩子可能的長相。
本文通過百度AI開放平臺的人臉融合功能來做一個簡單的試驗。
前期準備
賬號註冊
我們要使用百度AI開發平臺的功能,必須先註冊一個賬號。訪問 https://login.bce.baidu.com/ ,然後用你的百度賬號登錄就行。
登錄之後,在左邊的菜單欄依次選擇“產品服務 -> 人工智慧 -> 人臉識別”子菜單,進入到人臉識別的產品界面:
然後點擊“創建應用”,填寫“應用名稱”和“應用描述”即可創建應用:
然後返回應用列表,就可以看到你創建的應用:
創建完應用,你需要將 API Key 和 Secret Key 記下來,我們待會的代碼裡面會用到。
閱讀開發文檔
應用創建完後,我們需要知道怎麼調用百度的API來完成我們的試驗,所以我們需要閱讀官方文檔。我們需要做兩件事情:鑒權認
證和圖片融合。
鑒權認證文檔
如果要調用百度的 API 介面,必須先鑒權認證,也就是獲取應用 token 。獲取 token 的文檔地址為:https://ai.baidu.com/ai-
doc/FACE/5k37c1ti0。
文檔中對我們有用的幾處信息如下:
請求URL數據格式:
向授權服務地址https://aip.baidubce.com/oauth/2.0/token發送請求(推薦使用POST),併在URL中帶上以下參數:
•grant_type:必須參數,固定為client_credentials;
•client_id:必須參數,應用的API Key;
•client_secret:必須參數,應用的Secret Key;
請求舉例:
https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=Va5yQRHlA4Fq5eR3LT0vuXV4&client_secret=0rDSjzQ20XUj5itV6WRtznPQSzr5pVw2&
伺服器返回的JSON文本參數如下:
•access_token:要獲取的Access Token;
•expires_in:Access Token的有效期(秒為單位,一般為1個月);
•其他參數忽略,暫時不用;
返回舉例:
{ "refresh_token": "25.b55fe1d287227ca97aab219bb249b8ab.315360000.1798284651.282335-8574074", "expires_in": 2592000, "scope": "public wise_adapt", "session_key": "9mzdDZXu3dENdFZQurfg0Vz8slgSgvvOAUebNFzyzcpQ5EnbxbF+hfG9DQkpUVQdh4p6HbQcAiz5RmuBAja1JJGgIdJI", "access_token": "24.6c5e1ff107f0e8bcef8c46d3424a0e78.2592000.1485516651.282335-8574074", "session_secret": "dfac94a3489fe9fca7c3221cbf7525ff"}
圖片融合文檔
圖片融合文檔的地址為:https://ai.baidu.com/ai-doc/FACE/5k37c1ti0。
文檔中對我們有用的幾處信息如下:
請求註意事項:
•請求體格式化:Content-Type為application/json,通過json格式化請求體。
•Base64編碼:請求的圖片需經過Base64編碼,圖片的base64編碼指將圖片數據編碼成一串字元串,使用該字元串代替圖像地
址。您可以首先得到圖片的二進位,然後用Base64格式編碼即可。需要註意的是,圖片的base64編碼是不包含圖片頭的,如
data:image/jpg;base64。
•圖片格式:現支持PNG、JPG、JPEG、BMP,不支持GIF圖片。
請求示例:
•HTTP方法:POST
•請求URL:https://aip.baidubce.com/rest/2.0/face/v1/merge
•URL參數:access_token
•Header:Content-Type 為application/json
•Body中放置請求參數
返回示例:
Python學習交流Q群:906715085#### { "error_code": 0, "error_msg": "SUCCESS", "log_id": 1234567890123, "timestamp": 1533094576, "cached": 0, "result": { "merge_image": "iVBORw0KGgoAAAANSUhEUgAAAeoAAAHqCAYAAADLb..." } }
編碼
獲取 token
根據上面的文檔描述,我們調用鑒權介面的方法如下:
#獲取token def get_token(client_id, client_secret): url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials" params = {"client_id": client_id, "client_secret": client_secret} res = requests.get(url, params=params) result = res.json() return result['access_token']
這個介面很簡單,我們傳入我們應用的信息就可以直接獲得。
獲取圖片的 base64 編碼
我們調用百度人臉融合介面,需要傳入圖片的 base64 編碼,所以我們先要將圖片轉為 base64 格式,轉換方法如下:
#讀取圖片,轉換成base64 def read_pic(name): with open('./%s' % name, 'rb') as f: base64_data = base64.b64encode(f.read()) s = base64_data.decode() return s
我將圖片放在程式同級目錄下,方法傳入圖片文件名就可以。
調用人臉融合介面並保存結果
token 值和圖片的 base64 編碼準備好了,我們就可以來調用介面進行融合。根據官方的 API 文檔,我們的調用方法如下:
#融合圖片 def merge(token, template, target): url = 'https://aip.baidubce.com/rest/2.0/face/v1/merge' request_url = url + '?access_token=' + token params = { "image_template": { "image": template, "image_type": "BASE64", "quality_control": "NORMAL" }, "image_target": { "image": target, "image_type": "BASE64", "quality_control": "NORMAL" }, "merge_degree": "HIGH" } params = json.dumps(params) headers = {'content-type': 'application/json'} result = requests.post(request_url, data=params, headers=headers).json() if result['error_code'] == 0: res = result["result"]["merge_image"] down_pic(res) else: print(str(result['error_code'])+result['error_msg'])
參數中的 template 指的是模板圖片,target 指的是被融合圖片。也就是說將 target 圖片的人臉融合到 template 圖片的人臉中,最後輸出的圖片是以 template 圖片為模板的。
這裡有一個將介面返回的圖片轉存到本地的方法 down_pic ,其實現如下:
#下載圖片 def down_pic(data): imagedata = base64.b64decode(data) file = open('./result.jpg', "wb") file.write(imagedata)
我們把融合的圖片命名為 result.jpg ,存儲在程式同級目錄下。
主程式
主要的方法我們都完成了,下麵我們通過編寫主程式來測試我們的融合效果,代碼如下:
if __name__ == '__main__': girl = read_pic('girl.jpg') boy = read_pic('boy.jpg') token = get_token(API_KEY, SECRET_KEY) merge(token, boy, girl)
我這裡用一個男人的圖片和一個女人的圖片來做測試,以男人的圖片作為模板,兩張圖片都是從百度圖片搜索出來的。
男人的人臉圖片為:
女人的人臉圖片為:
融合後的人臉圖片為:
是不是很帥氣,這個可能是這兩個人未來兒子的模樣。接著,我們調換一下模板,我們以女人的圖片作為模板,看看他們未來女兒的模樣,結果如下:
本文通過調用百度AI開放平臺的人臉融合介面,來實現兩張正面人臉圖片的融合試驗。大家覺得融合的效果怎麼樣?我覺得如果這兩個人結婚,生男孩子會好看些呢!大家也可以把自己和另一半的自拍照拿來試驗一下,看看未來孩子長啥樣?當然,如果你是單身狗,那可以找個漂亮明星圖片來幻想一下。