[Update] Fix bug

open-mmlab · Jul 25, 2023 · 34a50de · 34a50de
1 parent 1dcd6fa
commit 34a50de
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmocr/models/textdet/detectors/mmdet_wrapper.py b/mmocr/models/textdet/detectors/mmdet_wrapper.py
@@ -138,6 +138,7 @@ def adapt_predictions(self, data: MMDET_SampleList,
                 # convert by text_repr_type
                 if self.text_repr_type == 'quad':
                     for j, poly in enumerate(filterd_polygons):
+                        poly = poly.reshape(-1, 2)
                         rect = cv2.minAreaRect(poly)
                         vertices = cv2.boxPoints(rect)
                         poly = vertices.flatten()