fix img_shape (#719)

open-mmlab · Feb 6, 2023 · 4922856 · 4922856
1 parent bff5160
commit 4922856
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmrotate/datasets/transforms/transforms.py b/mmrotate/datasets/transforms/transforms.py
@@ -151,7 +151,8 @@ def _transform_bboxes(self, results: dict) -> None:
 
     def _filter_invalid(self, results: dict) -> None:
         """Filter invalid data w.r.t `gt_bboxes`"""
-        height, width = results['img_shape']
+        # results['img_shape'] maybe (h,w,c) or (h,w)
+        height, width = results['img_shape'][:2]
         if 'gt_bboxes' in results:
             if len(results['gt_bboxes']) == 0:
                 return