import requests
import base64
class FindStr:
def __init__(self, file):
self.API_KEY = '???'
self.Secret_Key = '???'
self.file = file
self.token = None
def get_token(self):
host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s' % (
self.API_KEY, self.Secret_Key)
respones = requests.get(host)
if respones:
result = respones.json()
print(result)
self.token = result['access_token']
def find_str(self, findstr):
self.get_token()
request_url = 'https://aip.baidubce.com/rest/2.0/ocr/v1/general?access_token=%s' % (self.token)
f = open(self.file, 'rb')
img = base64.b64encode(f.read())
params = {'image': img}
headers = {'content-type': 'application/x-www-form-urlencoded'}
response = requests.post(request_url, data=params, headers=headers)
if response:
result = response.json()
print(result)
print(result['words_result'])
for item in result['words_result']:
print(item)
if item['words'] == findstr:
print(item['words'])
return item['location']
print('没找到')
return -1
if __name__ == '__main__':
fs = FindStr('test.bmp')
result = fs.find_str('你想要找的字符串')
print(result)
百度云识字:https://cloud.baidu.com/product/ocr