ONNX 模型库
返回模型

说明文档

<p align="center"> <img src="https://github.com/felixdittrich92/OnnxTR/raw/main/docs/images/logo.jpg" width="40%"> </p>

由 Onnxruntime 驱动的光学字符识别,无缝且易于使用

任务:检测

https://github.com/felixdittrich92/OnnxTR

使用示例:

>>> from onnxtr.io import DocumentFile
>>> from onnxtr.models import ocr_predictor, from_hub

>>> img = DocumentFile.from_images(['<image_path>'])
>>> # 从 Hub 加载你的模型
>>> model = from_hub('onnxtr/my-model')

>>> # 将其传递给预测器
>>> # 如果你的模型是识别模型:
>>> predictor = ocr_predictor(det_arch='db_mobilenet_v3_large',
>>>                           reco_arch=model)

>>> # 如果你的模型是检测模型:
>>> predictor = ocr_predictor(det_arch=model,
>>>                           reco_arch='crnn_mobilenet_v3_small')

>>> # 获取预测结果
>>> res = predictor(img)

Felix92/onnxtr-linknet-resnet18

作者 Felix92

↓ 61 ♥ 0

创建时间: 2024-08-16 06:34:08+00:00

更新时间: 2024-08-16 08:39:23+00:00

在 Hugging Face 上查看

文件 (4)

.gitattributes
README.md
config.json
model.onnx ONNX