| 12345678910111213141516 |
- import os
- from pathlib import Path
- def get_ocr_res(pipeline, input):
- """get ocr res"""
- ocr_res_list = []
- if isinstance(input, list):
- img = [im["img"] for im in input]
- elif isinstance(input, dict):
- img = input["img"]
- else:
- img = input
- for ocr_res in pipeline(img):
- ocr_res_list.append(ocr_res)
- return ocr_res_list
|