我要评论图片文字提取
需求需要，上传商品时不可上传不可私营产品，例如香烟、食盐等，以下仅供参考，如想放入生产环境，需要谨慎
import requests
import base64
def get_word_token():
url = 'https://aip.baidubce.com/oauth/2.0/token'
data = {
'grant_type': 'client_credentials', # 固定值
'client_id': 'Su6KSI9M7cTs9****A93dg3Z', # 在开放平台注册后所建应用的API Key
'client_secret': 'aCbgVGLLc0f****VD4LcqRwHZmf1Ee4E' # 所建应用的Secret Key
}
res = requests.post(url, data=data)
res = res.json()
print(res)
access_token = res['access_token']
return access_token
def distinguish_word():
"""提取图片上的文字进行判断"""
try:
dis_word_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic" # 通用文字识别接口(还可以提高识别精度有另外接口)
# 二进制方式打开图片文件
f = open('文字图片/c.jpg', 'rb')
word_img = base64.b64encode(f.read())
params = {"image":word_img, "language_type":"CHN_ENG"}
dis_token = get_word_token()
request_url = dis_word_url + "?access_token=" + dis_token
# print(request_url)
headers = {'content-type': 'text/html;charset=UTF-8'}
response = requests.post(request_url, data=params, headers=headers)
res = response.json()
print("请求信息:",res)
if response:
res = response.json()["words_result"]
mg = []
for i in res:
msg = i["words"]
mg.append(msg)
if any("烟" in s for s in mg):
print("香烟不可上传")
elif any("盐" in s for s in mg):
print("食盐不可上传")
except:
pass
if __name__ == '__main__':
distinguish_word()
您可能感兴趣的文章:
如您对本文有疑问或者有任何想说的，请 点击进行留言回复，万千网友为您解惑！
网友评论