Fix get_boxes_img_without_det in the python api (#150)

RapidAI · Jan 30, 2024 · d05b629 · d05b629
1 parent f686541
commit d05b629
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/rapidocr_onnxruntime/main.py b/python/rapidocr_onnxruntime/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