egullickson/motovaultpro
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Improve OCR process - replace Tesseract with PaddleOCR (#115) #122

Merged
egullickson merged 16 commits from issue-115-improve-ocr-paddleocr into main 2026-02-08 01:13:35 +00:00
Conversation 1 Commits 16 Files Changed 35 +19 -7
2 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 9a2b12c5dc - Show all commits

13
ocr/app/engines/paddle_engine.py
View File

@@ -56,8 +56,9 @@ class PaddleOcrEngine(OcrEngine):
"""Run PaddleOCR on image bytes. """Run PaddleOCR on image bytes.
PaddleOCR v3.x ``predict()`` returns an iterator of result objects. PaddleOCR v3.x ``predict()`` returns an iterator of result objects.
Each result has a ``res`` dict with ``dt_polys``, ``rec_texts``, Each result's ``.json`` property returns a dict. The OCR fields
and ``rec_scores``. (``dt_polys``, ``rec_texts``, ``rec_scores``) may be at the top
level or nested under a ``"res"`` key depending on the version.
""" """
ocr = self._get_ocr() ocr = self._get_ocr()
@@ -78,7 +79,13 @@ class PaddleOcrEngine(OcrEngine):
engine_name=self.name, engine_name=self.name,
) )
res = results[0].json raw = results[0].json
# Unwrap nested "res" key if present (save_to_json format)
res = raw.get("res", raw) if isinstance(raw, dict) else raw
logger.debug(
"PaddleOCR result keys: %s",
list(res.keys()) if isinstance(res, dict) else type(res).__name__,
)
dt_polys = res.get("dt_polys", []) dt_polys = res.get("dt_polys", [])
rec_texts = res.get("rec_texts", []) rec_texts = res.get("rec_texts", [])
rec_scores = res.get("rec_scores", []) rec_scores = res.get("rec_scores", [])

13
ocr/tests/test_engine_abstraction.py
View File

@@ -44,12 +44,17 @@ def _make_result(
def _mock_paddle_result( def _mock_paddle_result(
dt_polys: list, rec_texts: list[str], rec_scores: list[float] dt_polys: list, rec_texts: list[str], rec_scores: list[float]
) -> MagicMock: ) -> MagicMock:
"""Create a mock PaddleOCR v3.x predict() result object.""" """Create a mock PaddleOCR v3.x predict() result object.
Wraps data under ``"res"`` key to match save_to_json format.
"""
result = MagicMock() result = MagicMock()
result.json = { result.json = {
"dt_polys": dt_polys, "res": {
"rec_texts": rec_texts, "dt_polys": dt_polys,
"rec_scores": rec_scores, "rec_texts": rec_texts,
"rec_scores": rec_scores,
}
} }
return result return result