Albumentations default key mapping mismatch #3179
Comments
|
Hi @Alias-z, |
|
Sure! See PR: #3195 |
xiexinch
added a commit
that referenced
this issue
Jul 14, 2023
## Modification Fix Albumentations default key mapping mismatch as mentioned in [issue # 3179](#3179) by changing `self.keymap_to_albu = { 'img': 'image', 'gt_masks': 'masks'}` to `self.keymap_to_albu = { 'img': 'image', 'gt_seg_map': 'mask'}` ## Use cases (Optional) Example albu config ``` crop_size = (512, 512) albu_train_transforms = [ dict( type='PadIfNeeded', min_height=crop_size[0]*2, min_width=crop_size[1]*2, border_mode=0, always_apply=True), dict(type='Flip', always_apply=True), dict(type='Rotate', limit=(-180, 180), interpolation=4, always_apply=True), dict(type='RandomScale', scale_limit=0.1, interpolation=4, always_apply=True), dict( type='ElasticTransform', alpha=20, sigma=15, interpolation=4, border_mode=0, mask_value=(0, 0, 0), approximate=True, same_dxdy=True, p=0.8), dict(type='ColorJitter', brightness=0.2, contrast=0.1, saturation=0.2, hue=0.2, always_apply=True), dict(type='AdvancedBlur', p=0.5), dict(type='CenterCrop', height=crop_size[0], width=crop_size[1], always_apply=True) ] ``` Example training pipeline without specifying `keymap` ``` train_pipeline = [ dict(type='LoadImageFromFile'), dict(type='LoadAnnotations', reduce_zero_label=False), dict( type='Albu', transforms=albu_train_transforms, ), dict(type='Resize', scale=crop_size, keep_ratio=False, interpolation='lanczos'), dict(type='PackSegInputs') ] ``` Example viz_dataset before the issue fixing  Example viz_dataset after the issue fixing  --------- Co-authored-by: xiexinch <xiexinch@outlook.com>
nahidnazifi87
pushed a commit
to nahidnazifi87/mmsegmentation_playground
that referenced
this issue
Apr 5, 2024
## Modification Fix Albumentations default key mapping mismatch as mentioned in [issue # 3179](open-mmlab#3179) by changing `self.keymap_to_albu = { 'img': 'image', 'gt_masks': 'masks'}` to `self.keymap_to_albu = { 'img': 'image', 'gt_seg_map': 'mask'}` ## Use cases (Optional) Example albu config ``` crop_size = (512, 512) albu_train_transforms = [ dict( type='PadIfNeeded', min_height=crop_size[0]*2, min_width=crop_size[1]*2, border_mode=0, always_apply=True), dict(type='Flip', always_apply=True), dict(type='Rotate', limit=(-180, 180), interpolation=4, always_apply=True), dict(type='RandomScale', scale_limit=0.1, interpolation=4, always_apply=True), dict( type='ElasticTransform', alpha=20, sigma=15, interpolation=4, border_mode=0, mask_value=(0, 0, 0), approximate=True, same_dxdy=True, p=0.8), dict(type='ColorJitter', brightness=0.2, contrast=0.1, saturation=0.2, hue=0.2, always_apply=True), dict(type='AdvancedBlur', p=0.5), dict(type='CenterCrop', height=crop_size[0], width=crop_size[1], always_apply=True) ] ``` Example training pipeline without specifying `keymap` ``` train_pipeline = [ dict(type='LoadImageFromFile'), dict(type='LoadAnnotations', reduce_zero_label=False), dict( type='Albu', transforms=albu_train_transforms, ), dict(type='Resize', scale=crop_size, keep_ratio=False, interpolation='lanczos'), dict(type='PackSegInputs') ] ``` Example viz_dataset before the issue fixing  Example viz_dataset after the issue fixing  --------- Co-authored-by: xiexinch <xiexinch@outlook.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Describe the bug
Masks would not transform along with input images using the default key mapping
self.keymap_to_albu = { 'img': 'image', 'gt_masks': 'masks'}Reproduction
Using default setting:
Using
keymap={'img': 'image', 'gt_seg_map': 'mask'}Bug fix
keymap={'img': 'image', 'gt_seg_map': 'mask'}The text was updated successfully, but these errors were encountered: