Fixed the PR #150 in rapidocr_vino and rapidocr_paddle module

RapidAI · Jan 30, 2024 · a2e11df · a2e11df
1 parent d05b629
commit a2e11df
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 2 deletions.
diff --git a/python/rapidocr_openvino/main.py b/python/rapidocr_openvino/main.py
@@ -126,7 +126,7 @@ def auto_text_det(
 
     def get_boxes_img_without_det(self, h, w):
         x0, y0, x1, y1 = 0, 0, w, h
-        dt_boxes = np.array([[x0, y0], [x1, y0], [x1, y1], [x0, y1]])
+        dt_boxes = np.float32([[x0, y0], [x1, y0], [x1, y1], [x0, y1]])
         dt_boxes = dt_boxes[np.newaxis, ...]
         return dt_boxes
 

diff --git a/python/rapidocr_paddle/main.py b/python/rapidocr_paddle/main.py
@@ -126,7 +126,7 @@ def auto_text_det(
 
     def get_boxes_img_without_det(self, h, w):
         x0, y0, x1, y1 = 0, 0, w, h
-        dt_boxes = np.array([[x0, y0], [x1, y0], [x1, y1], [x0, y1]])
+        dt_boxes = np.float32([[x0, y0], [x1, y0], [x1, y1], [x0, y1]])
         dt_boxes = dt_boxes[np.newaxis, ...]
         return dt_boxes
 

diff --git a/python/tests/test_files/test_without_det.png b/python/tests/test_files/test_without_det.png
diff --git a/python/tests/test_ort.py b/python/tests/test_ort.py
@@ -20,6 +20,13 @@
 package_name = "rapidocr_onnxruntime"
 
 
+def test_without_det():
+    img_path = tests_dir / "test_without_det.png"
+    result, _ = engine(img_path)
+    assert result[0][1] == "の持、持场所无。"
+    assert len(result) == 1
+
+
 def test_only_det():
     result, _ = engine(img_path, use_det=True, use_cls=False, use_rec=False)
 

diff --git a/python/tests/test_paddle.py b/python/tests/test_paddle.py
@@ -20,6 +20,13 @@
 package_name = "rapidocr_paddle"
 
 
+def test_without_det():
+    img_path = tests_dir / "test_without_det.png"
+    result, _ = engine(img_path)
+    assert result[0][1] == "の持、持场所无。"
+    assert len(result) == 1
+
+
 def test_only_det():
     result, _ = engine(img_path, use_det=True, use_cls=False, use_rec=False)
 

diff --git a/python/tests/test_vino.py b/python/tests/test_vino.py
@@ -20,6 +20,13 @@
 package_name = "rapidocr_openvino"
 
 
+def test_without_det():
+    img_path = tests_dir / "test_without_det.png"
+    result, _ = engine(img_path)
+    assert result[0][1] == "の持、持场所无。"
+    assert len(result) == 1
+
+
 def test_only_det():
     result, _ = engine(img_path, use_det=True, use_cls=False, use_rec=False)