diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py index 702bdea07161a135a5b46b61361d71887fbf943f..a9516d5f432e3932dd5a6a033ca8f725323b4a3f 100644 --- a/mmdet/datasets/pipelines/transforms.py +++ b/mmdet/datasets/pipelines/transforms.py @@ -833,9 +833,8 @@ class Albu(object): results[label] = np.array( [results[label][i] for i in results['idx_mapper']]) if 'masks' in results: - results['masks'] = [ - results['masks'][i] for i in results['idx_mapper'] - ] + results['masks'] = np.array( + [results['masks'][i] for i in results['idx_mapper']]) if (not len(results['idx_mapper']) and self.skip_img_without_anno):