-
Notifications
You must be signed in to change notification settings - Fork 0
/
AIp-cor.py
40 lines (29 loc) · 1.16 KB
/
AIp-cor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import time
from aip import AipOcr
class Aiocr(object):
def __init__(self, app_id, api_key, secret_key):
self._app_id = app_id
self._api_key = api_key
self._secret_key = secret_key
self._client = AipOcr(app_id, api_key, secret_key)
def proc_form(self, image):
image = self.get_file_content(image)
return self._client.tableRecognitionAsync(image);
def proc_from_res(self, req_id):
options = {"result_type": "excel"}
return self._client.getTableRecognitionResult(req_id, options)
def get_file_content(self, file_path):
with open(file_path, 'rb') as fp:
return fp.read()
if __name__ == '__main__':
APP_ID = '11635479'
API_KEY = 'y52YctCoPhY8KRwtzoFliKsU'
SECRET_KEY = 'GawZoudCnmRhqq1lDbqkpgky3rpsmk8G'
ai_ocr = Aiocr(APP_ID, API_KEY, SECRET_KEY)
res = ai_ocr.proc_form('./image/1.jpg')
for req in res['result']:
resp = ai_ocr.proc_from_res(req['request_id'])
while not ('error_code' in resp) and not resp['result']['ret_code'] == 3:
time.sleep(10)
resp = ai_ocr.proc_from_res(req['request_id'])
print(resp)