1. 程式人生 > 其它 >Python人臉融合告訴你,你的下一代長啥樣

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開放平臺的人臉融合介面,來實現兩張正面人臉圖片的融合試驗。大家覺得融合的效果怎麼樣?我覺得如果這兩個人結婚,生男孩子會好看些呢!大家也可以把自己和另一半的自拍照拿來試驗一下,看看未來孩子長啥樣?當然,如果你是單身狗,那可以找個漂亮明星圖片來幻想一下。