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