[Fix] Fix a bug in MMDetWrapper

open-mmlab · Sep 16, 2022 · 93d883e · 93d883e
1 parent 87f15b3
commit 93d883e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmocr/models/textdet/detectors/mmdet_wrapper.py b/mmocr/models/textdet/detectors/mmdet_wrapper.py
@@ -137,11 +137,11 @@ def adapt_predictions(self, data: MMDET_SampleList,
                     keep_idx.append(poly_idx)
                 # convert by text_repr_type
                 if self.text_repr_type == 'quad':
-                    for i, poly in enumerate(filterd_polygons):
+                    for j, poly in enumerate(filterd_polygons):
                         rect = cv2.minAreaRect(poly)
                         vertices = cv2.boxPoints(rect)
                         poly = vertices.flatten()
-                        filterd_polygons[i] = poly
+                        filterd_polygons[j] = poly
 
                 data_samples[i].pred_instances.polygons = filterd_polygons
                 data_samples[i].pred_instances.scores = torch.FloatTensor(