菜品识别 API调用

    #get_access_token.py
    #获取access_token
 1 import requests



def GetToken(API_KEY,SECRET_KEY):
      url = 'https://aip.baidubce.com/oauth/2.0/token?' 
            'grant_type=client_credentials&' 
            'client_id='+API_KEY+'&' 
            'client_secret='+SECRET_KEY
      response=requests.get(url)
      print('access_token=',response.json()['access_token'])
      return response.json()['access_token']

调用API

import requests
import base64
from get_access_token import GetToken
import os

""" 填入参数 """
API_KEY = 'your API_KEY'
SECRET_KEY = 'your SECRET_KEY'
url = "https://aip.baidubce.com/rest/2.0/image-classify/v2/dish"

def DishRecg(url,data,access_token,folder):
    header = {
        'Content-Type': 'application/x-www-form-urlencoded'
    }
    request_url = url + "?access_token=" + access_token
    response=requests.post(url=request_url,data=data,headers=header).json()
    result=response['result'][0]
    print(folder,
          '	菜品为：',result['name'],
          '	置信度：', result['probability'],
          '	卡路里：', result['calorie'])


def image_base64(imagepath):
    # 二进制方式打开图片文件
    f = open(imagepath, 'rb')
    img = base64.b64encode(f.read())
    return img


if __name__ == "__main__":
    data={}
    access_token = GetToken(API_KEY, SECRET_KEY)
    dish_path = os.getcwd()
    dish_path = dish_path + '/dish'
    folderlist = os.listdir(dish_path)
    for folder in folderlist:
        imagepath=dish_path + '/' + folder
        image=image_base64(imagepath)
        data['image']=str(image,'utf-8')
        DishRecg(url,data,access_token,folder)