菜品识别 API调用
#get_access_token.py
#获取access_token
1 import requests 2 3 4 5 def GetToken(API_KEY,SECRET_KEY): 6 url = 'https://aip.baidubce.com/oauth/2.0/token?' \ 7 'grant_type=client_credentials&' \ 8 'client_id='+API_KEY+'&' \ 9 'client_secret='+SECRET_KEY 10 response=requests.get(url) 11 print('access_token=',response.json()['access_token']) 12 return response.json()['access_token']
调用API
1 import requests 2 import base64 3 from get_access_token import GetToken 4 import os 5 6 """ 填入参数 """ 7 API_KEY = 'your API_KEY' 8 SECRET_KEY = 'your SECRET_KEY' 9 url = "https://aip.baidubce.com/rest/2.0/image-classify/v2/dish" 10 11 def DishRecg(url,data,access_token,folder): 12 header = { 13 'Content-Type': 'application/x-www-form-urlencoded' 14 } 15 request_url = url + "?access_token=" + access_token 16 response=requests.post(url=request_url,data=data,headers=header).json() 17 result=response['result'][0] 18 print(folder, 19 '\t菜品为：',result['name'], 20 '\t置信度：', result['probability'], 21 '\t卡路里：', result['calorie']) 22 23 24 def image_base64(imagepath): 25 # 二进制方式打开图片文件 26 f = open(imagepath, 'rb') 27 img = base64.b64encode(f.read()) 28 return img 29 30 31 if __name__ == "__main__": 32 data={} 33 access_token = GetToken(API_KEY, SECRET_KEY) 34 dish_path = os.getcwd() 35 dish_path = dish_path + '/dish' 36 folderlist = os.listdir(dish_path) 37 for folder in folderlist: 38 imagepath=dish_path + '/' + folder 39 image=image_base64(imagepath) 40 data['image']=str(image,'utf-8') 41 DishRecg(url,data,access_token,folder)
