OCR_Processo.py 789 B

12345678910111213141516171819
  1. from .... import requests,cv2,time,random,os,tempfile,logger,re
  2. class OCRProcesso:
  3. def __init__(self):
  4. ...
  5. def ocr(self,imgs):
  6. # 创建临时目录
  7. temp_dir = tempfile.mkdtemp()
  8. temp_image_path = os.path.join(temp_dir, f'temp_image_{time.time()}_{random.randint(0, 1000)}.png')
  9. logger.info(f'文件地址 {temp_image_path}')
  10. cv2.imwrite(temp_image_path, imgs)
  11. r = requests.post(
  12. 'http://183.195.216.54:9587/ocr', files={'image': (temp_image_path, open(temp_image_path, 'rb'), 'image/png')},
  13. )
  14. ocr_out = r.json()['results']
  15. ocr_out = ''.join( [item['text'] for item in ocr_out])
  16. if ocr_out == 'ELSEVTER':
  17. ocr_out = ''
  18. logger.debug(ocr_out)
  19. return ocr_out