呼叫百度人臉檢測API實現簡單的顏值檢測
阿新 • • 發佈:2019-01-26
通過百度人工智慧平臺中的人臉檢測模組,實現簡單的人臉檢測,百度人工智慧平臺免費註冊,人臉檢測模組免費使用。相關API文件在:http://ai.baidu.com/docs#/Face-Detect-V3/top,依照API文件就可以寫出一個簡單的檢測工具。程式碼如下:
import base64 import json import requests class BaiduPicIndentify: def __init__(self,img): self.AK = "換成你自己的AK" self.SK = "換成你自己的SK" self.img_src = img self.headers = { "Content-Type": "application/json; charset=UTF-8" } def get_accessToken(self): host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + self.AK + '&client_secret=' + self.SK response = requests.get(host, headers=self.headers) json_result = json.loads(response.text) return json_result['access_token'] def img_to_BASE64(slef,path): with open(path,'rb') as f: base64_data = base64.b64encode(f.read()) return base64_data def detect_face(self): # 人臉檢測與屬性分析 img_BASE64 = self.img_to_BASE64(self.img_src) request_url = "https://aip.baidubce.com/rest/2.0/face/v3/detect" post_data = { "image": img_BASE64, "image_type": "BASE64", "face_field": "gender,age,beauty,gender,race,expression", "face_type": "LIVE" } access_token = self.get_accessToken() request_url = request_url + "?access_token=" + access_token response = requests.post(url=request_url, data=post_data, headers=self.headers) json_result = json.loads(response.text) if json_result['error_msg']!='pic not has face': print("圖片中包含人臉數:", json_result['result']['face_num']) print("圖片中包含人物年齡:", json_result['result']['face_list'][0]['age']) print("圖片中包含人物顏值評分:", json_result['result']['face_list'][0]['beauty']) print("圖片中包含人物性別:", json_result['result']['face_list'][0]['gender']['type']) print("圖片中包含人物種族:", json_result['result']['face_list'][0]['race']['type']) print("圖片中包含人物表情:", json_result['result']['face_list'][0]['expression']['type']) if __name__=='__main__': img_src=input('請輸入需要檢測的本地圖片路徑:') baiduDetect = BaiduPicIndentify(img_src) baiduDetect.detect_face()
輸入的圖片1.JPG及其檢測結果如下:
檢測結果: SUCCESS
圖片中包含人臉數: 1
圖片中包含人物年齡: 23
圖片中包含人物顏值評分: 60.61511993
圖片中包含人物性別: female
圖片中包含人物種族: yellow
圖片中包含人物表情: none
輸入圖片2及其檢測結果如下:
檢測結果: SUCCESS
圖片中包含人臉數: 1
圖片中包含人物年齡: 22
圖片中包含人物顏值評分: 77.99679565
圖片中包含人物性別: female
圖片中包含人物種族: yellow
圖片中包含人物表情: smile
輸入圖片3及其檢測結果如下:
檢測結果: pic not has face