NeuHub图像垃圾分类api和百度图像识别api

京东

NeuHub图像垃圾分类申请：http://neuhub.jd.com/gwtest/init/242

文档：https://aidoc.jd.com/image/garbageClassification.html

import base64
import wx_sdk  #我是将wx_sdk.py移同当前文件夹了
import json

url = \'https://aiapi.jd.com/jdai/garbageImageSearch\'
f = open(\'nfsq.jpg\', \'rb\')
#转成base64
image_base64 = str(base64.b64encode(f.read()), encoding=\'utf-8\')
#自己xjb并凑的
bodys = "{\"cityId\"" + \':\' + "\"310000\"" + ", " + "\"imgBase64\"" + \':\' + "\"" + image_base64 + "\"" "}"
#bodyStr = \'{ "cityId":"310000", "imgBase64":"image_base64"}\'

params = { 
    \'appkey\' : \'你的appkey\',
    \'secretkey\' : \'你的secretkey\'}
response = wx_sdk.wx_post_req(url, params, bodyStr=bodys)
#print(response.text)

#将json格式转成字典
result = json.loads(response.text)
#输出自己想要的一些信息
for key in result["result"]["garbage_info"]:
    if(key["confidence"] > 0.5):  #只输出置信度超过0.5的，官方建议为0.7
        print(key["cate_name"], key["confidence"], key["garbage_name"])

百度

百度图像识别api :https://ai.baidu.com/docs#/ImageClassify-API/ebc492b1

1. 安装百度api

pip3 install baidu-aip

2. 代码

from aip import AipImageClassify

""" 你的 APPID AK SK """
APP_ID = \'你的APP_ID\'
API_KEY = \'你的APP_KEY\'
SECRET_KEY = \'你的SECRET_KEY\'
client = AipImageClassify(APP_ID, API_KEY, SECRET_KEY)

def get_file_content(filePath):
    with open(filePath, \'rb\') as fp:
        return fp.read()
image = get_file_content(\'nfsq.jpg\')


#返回百科信息的结果数，默认为0，不返回；2为返回前2个结果的百科信息，以此类推。
options = {}
options["baike_num"] = 5

""" 带参数调用通用物体识别 """
result  = client.advancedGeneral(image, options)
# print(result)
result_num = result[\'result_num\']
for i in range(0, result_num):
    print(result[\'result\'][i][\'keyword\'])

另一种使用api的方式是使用access_token

//检测图像中的主体位置，通用物体和场景识别的高级版是收费的？

1. 获取access_token

from urllib import request
import ssl
import json
gcontext = ssl.SSLContext(ssl.PROTOCOL_TLSv1)
# client_id 为官网获取的AK， client_secret 为官网获取的SK
host = \'https://aip.baidubce.com/oauth/2.0/token?grant_\' \
       \'type=client_credentials&client_id=你的AK&client_secret=你的SKreq = request.Request(host)
response = request.urlopen(req, context=gcontext).read().decode(\'UTF-8\')
result = json.loads(response)
if (result):
    print(result[\'access_token\'])

2. 将图片用base64编码

import base64
f = open(\'tiger.jpg\', \'rb\')
img = base64.b64encode(f.read())
print(img)

3. 调用api

import requests
import base64
host = \'https://aip.baidubce.com/rest/2.0/image-classify/v1/object_detect\'
headers={
   \'Content-Type\':\'application/x-www-form-urlencoded\'
}
access_token= \'xxx\'  #步骤1中获得的token
host=host+\'?access_token=\'+access_token

f = open(\'destop.jpg\', \'rb\')
img = base64.b64encode(f.read())
# print(img)

data={}
data[\'access_token\']=access_token
data[\'image\'] =img
res = requests.post(url=host,headers=headers,data=data)
req=res.json()
print(req[\'result\'])

参考链接：

1. https://blog.csdn.net/cool_bot/article/details/90465167

2. https://blog.csdn.net/qq_40484582/article/details/82054009