We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 580f63e commit 20fa129Copy full SHA for 20fa129
mmseg/datasets/transforms/transforms.py
@@ -2206,10 +2206,7 @@ def __init__(self,
2206
self.aug = Compose([self.albu_builder(t) for t in self.transforms])
2207
2208
if not keymap:
2209
- self.keymap_to_albu = {
2210
- 'img': 'image',
2211
- 'gt_masks': 'masks',
2212
- }
+ self.keymap_to_albu = {'img': 'image', 'gt_seg_map': 'mask'}
2213
else:
2214
self.keymap_to_albu = copy.deepcopy(keymap)
2215
self.keymap_back = {v: k for k, v in self.keymap_to_albu.items()}
0 commit comments