[Fix] Albumentations default key mapping mismatch #3195
Conversation
fix Albumentations default key mapping mismatch
|
|
|
Hi @Alias-z, We'd like to express our appreciation for your valuable contributions to the mmsegmentation. Your efforts have significantly aided in enhancing the project's quality. If you're on WeChat, we'd also love for you to join our community there. Just add our assistant using the WeChat ID: openmmlabwx. When sending the friend request, remember to include the remark "mmsig + Github ID". Thanks again for your awesome contribution, and we're excited to have you as part of our community! |
## 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>
Modification
Fix Albumentations default key mapping mismatch as mentioned in issue # 3179 by changing
self.keymap_to_albu = { 'img': 'image', 'gt_masks': 'masks'}toself.keymap_to_albu = { 'img': 'image', 'gt_seg_map': 'mask'}Use cases (Optional)
Example albu config
Example training pipeline without specifying
keymapExample viz_dataset before the issue fixing
Example viz_dataset after the issue fixing
