mtfyun/rapidocr
1
0 Fork 0
mirror of https://github.com/RapidAI/RapidOCR.git synced 2026-11-11 05:20:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support PP-OCRv6 PyTor as the inference engine

Browse Source
This commit is contained in:
SWHL committed 2026-07-02 17:54:27 +08:00
1 parent c18432567f
commit 370f01d0a8
11 files changed
+1100 -288
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

python/tests/test_engine.py
+14 -2
View File
@@ -143,7 +143,13 @@ def test_engine_tensorrt():
@mark.parametrize(
"engine_type",
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE],
[
EngineType.ONNXRUNTIME,
EngineType.OPENVINO,
EngineType.PADDLE,
EngineType.TORCH,
EngineType.MNN,
],
)
def test_ppocrv6_det_small(engine_type):
engine = RapidOCR(
@@ -161,7 +167,13 @@ def test_ppocrv6_det_small(engine_type):
@mark.parametrize(
"engine_type",
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE],
[
EngineType.ONNXRUNTIME,
EngineType.OPENVINO,
EngineType.PADDLE,
EngineType.TORCH,
EngineType.MNN,
],
)
def test_ppocrv6_rec_small(engine_type):
engine = RapidOCR(