Fixed mask shape after Albu postprocess (#11280)

open-mmlab · Dec 27, 2023 · b98f372 · b98f372
1 parent e5f9f35
commit b98f372
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mmdet/datasets/transforms/transforms.py b/mmdet/datasets/transforms/transforms.py
@@ -1766,8 +1766,10 @@ def _postprocess_results(
                     results['masks'] = np.array(
                         [results['masks'][i] for i in results['idx_mapper']])
                     results['masks'] = ori_masks.__class__(
-                        results['masks'], ori_masks.height, ori_masks.width)
-
+                        results['masks'],
+                        results['masks'][0].shape[0],
+                        results['masks'][0].shape[1],
+                    )
                 if (not len(results['idx_mapper'])
                         and self.skip_img_without_anno):
                     return None