Python人臉融合告訴你,你的下一代長啥樣
前言
和換臉演算法的簡單粗暴相比,人臉融合演算法就要彈性的多。顧名思義,人臉融合是對兩張人臉進行融合處理,生成的人臉同時具
備兩張人臉的外貌特徵。人臉融合有什麼實際意義呢?一個簡單的應用就是用父母雙方的臉部圖片融合,得到未來孩子可能的長相。
本文通過百度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開放平臺的人臉融合介面,來實現兩張正面人臉圖片的融合試驗。大家覺得融合的效果怎麼樣?我覺得如果這兩個人結婚,生男孩子會好看些呢!大家也可以把自己和另一半的自拍照拿來試驗一下,看看未來孩子長啥樣?當然,如果你是單身狗,那可以找個漂亮明星圖片來幻想一下。