Skip to content

GitLab

Commit e60d34af authored by Cao Yuhang's avatar Cao Yuhang Committed by Kai Chen
Browse files

Fix albu mask format bug (#1818)

parent c67cf3c4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
mmdet/datasets/pipelines/transforms.py
View file @ e60d34af
...@@ -833,9 +833,8 @@ class Albu(object): ...@@ -833,9 +833,8 @@ class Albu(object):
results[label] = np.array( results[label] = np.array(
[results[label][i] for i in results['idx_mapper']]) [results[label][i] for i in results['idx_mapper']])
if 'masks' in results: if 'masks' in results:
results['masks'] = [ results['masks'] = np.array(
results['masks'][i] for i in results['idx_mapper'] [results['masks'][i] for i in results['idx_mapper']])
]
if (not len(results['idx_mapper']) if (not len(results['idx_mapper'])
and self.skip_img_without_anno): and self.skip_img_without_anno):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment