Spaces:
Running
Running
File size: 2,172 Bytes
a383d0e |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import cv2
import os
import requests
import json
from base64 import b64encode
import time
def Google_OCR_makeImageData(imgpath):
with open(imgpath, 'rb') as f:
ctxt = b64encode(f.read()).decode()
img_req = {
'image': {
'content': ctxt
},
'features': [{
'type': 'DOCUMENT_TEXT_DETECTION',
# 'type': 'TEXT_DETECTION',
'maxResults': 1
}]
}
return json.dumps({"requests": img_req}).encode()
def ocr_detection_google(imgpath):
start = time.perf_counter()
url = 'https://vision.googleapis.com/v1/images:annotate'
api_key = 'AIzaSyDUc4iOUASJQYkVwSomIArTKhE2C6bHK8U' # *** Replace with your own Key ***
imgdata = Google_OCR_makeImageData(imgpath)
response = requests.post(url,
data=imgdata,
params={'key': api_key},
headers={'Content_Type': 'application/json'})
# print('*** Text Detection Time Taken:%.3fs ***' % (time.perf_counter() - start))
print("*** Please replace the Google OCR key at detect_text/ocr.py line 28 with your own (apply in https://cloud.google.com/vision) ***")
response_json = response.json()
if 'error' in response_json:
error_msg = response_json['error']
if 'BILLING_DISABLED' in str(error_msg):
raise Exception("Google Vision API requires billing to be enabled. Please:\n"
"1. Visit https://console.developers.google.com/billing/enable?project=718250946490\n"
"2. Enable billing for your project\n"
"3. Wait a few minutes for changes to propagate\n"
"4. Or use PaddleOCR instead by setting method='paddle'")
else:
raise Exception(f"Google Vision API error: {error_msg}")
if 'responses' not in response_json:
raise Exception(response_json)
if response_json['responses'] == [{}]:
# No Text
return None
else:
return response_json['responses'][0]['textAnnotations'][1:]
|