跳转至

模型列表

针对PaddleOCR已经发布的常用模型,我们这里已经做了统一转换和汇总,包括PaddlePaddle格式、ONNX格式和PyTorch格式。

所有模型目前托管在魔搭社区上。

所有模型汇总

请移步以下链接查看:

default_model.yaml

使用方式

以上模型可直接通过字段指定,程序会自动下载使用。

import cv2

from rapidocr import RapidOCR, VisRes

engine = RapidOCR(
    params={"Global.lang_det": "ch_mobile", "Global.lang_rec": "ch_mobile"}
)
vis = VisRes()

image_path = "tests/test_files/ch_en_num.jpg"
with open(image_path, "rb") as f:
    img = f.read()

result = engine(img)
print(result)
print(result.elapse)

vis_img = vis(img, result.boxes, result.txts, result.scores)
cv2.imwrite("vis.png", vis_img)

评论