feat: Improve OCR process - replace Tesseract with PaddleOCR (#115) #122
@@ -78,7 +78,7 @@ class PaddleOcrEngine(OcrEngine):
|
|||||||
engine_name=self.name,
|
engine_name=self.name,
|
||||||
)
|
)
|
||||||
|
|
||||||
res = results[0].res
|
res = results[0].json
|
||||||
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", [])
|
||||||
|
|||||||
@@ -46,7 +46,7 @@ def _mock_paddle_result(
|
|||||||
) -> MagicMock:
|
) -> MagicMock:
|
||||||
"""Create a mock PaddleOCR v3.x predict() result object."""
|
"""Create a mock PaddleOCR v3.x predict() result object."""
|
||||||
result = MagicMock()
|
result = MagicMock()
|
||||||
result.res = {
|
result.json = {
|
||||||
"dt_polys": dt_polys,
|
"dt_polys": dt_polys,
|
||||||
"rec_texts": rec_texts,
|
"rec_texts": rec_texts,
|
||||||
"rec_scores": rec_scores,
|
"rec_scores": rec_scores,
|
||||||
|
|||||||
Reference in New Issue
Block a user