pp_ocr.py

import tempfile
import os

import paddlehub as hub
from PIL import Image

pp_ocrv3 = hub.Module(name="ch_pp-ocrv3")

def inference_img(img):
    with tempfile.TemporaryDirectory() as tempdir_name:
        pp_ocrv3.recognize_text(images=[img], use_gpu=False, output_dir=tempdir_name, visualization=True)
        result_names = os.listdir(tempdir_name)
        result_image = Image.open(os.path.join(tempdir_name, result_names[0]))
        return result_image
    
def inference_json(img):
        results = pp_ocrv3.recognize_text(images=[img], use_gpu=False, visualization=False)
        return results