utils.py 374 B

12345678910111213141516
  1. import os
  2. from pathlib import Path
  3. def get_ocr_res(pipeline, input):
  4. """get ocr res"""
  5. ocr_res_list = []
  6. if isinstance(input, list):
  7. img = [im["img"] for im in input]
  8. elif isinstance(input, dict):
  9. img = input["img"]
  10. else:
  11. img = input
  12. for ocr_res in pipeline(img):
  13. ocr_res_list.append(ocr_res)
  14. return ocr_res_list