模型列表
针对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)
|