Closed gih0109 closed 1 year ago
mmsegmentation 에 없는 augmentation 을 추가합니다.
여러 Augmentation 을 사용할 수 있는 Albumentation Package 를 mmsegmentation 에 사용가능하게 합니다.
aug 목록은 아래 링크에서 확인가능합니다. https://albumentations.ai/docs/getting_started/transforms_and_targets/
aug 는 아래 링크에서 테스트 해볼 수 있습니다. https://yoyoyo-yo-albumentations-sim-sl-app-qevcnn.streamlit.app/
dataset config 내 train_pipeline 에 다음과 같은 코드를 넣어주면 됩니다.
train_pipeline = [ ... dict(type='Albu', transforms=[ # 해당 부분 수정 dict( type='ShiftScaleRotate', shift_limit=0.0625, scale_limit=0.0, rotate_limit=0, interpolation=1, p=0.5), dict( type='RandomBrightnessContrast', brightness_limit=[0.1, 0.3], contrast_limit=[0.1, 0.3], p=0.2), dict(type='ChannelShuffle', p=0.1), dict( type='OneOf', transforms=[ dict(type='Blur', blur_limit=3, p=1.0), dict(type='MedianBlur', blur_limit=3, p=1.0) ], p=0.1), ], keymap=dict(img='image', gt_semantic_seg='mask'), update_pad_shape=False, ), ... ]
## Change Log - transforms_custom.py 에 Albumentation 추가 ## To Reviewer mmsegmentation/tools/browse_dataset.py 를 통해 제대로 동작하는지 확인할 수 있습니다. - 다음 명령어로 실행 가능합니다. ```python python tools/browse_dataset.py {dataset_config} --output-dir={output_dir}
예시
python tools/browse_dataset.py config/dataset/test_config.py --output-dir='./output'
버그 발견시 알려주세요
확인했습니다. 감사합니다!
streamlit 링크도 그렇고 browse_dataset.py도 그렇고 신기하네요.
많이 배워갑니다!
잘 동작하는 것 확인했습니다~ 고생하셨습니다!
Overview
mmsegmentation 에 없는 augmentation 을 추가합니다.
Albumentation 추가
여러 Augmentation 을 사용할 수 있는 Albumentation Package 를 mmsegmentation 에 사용가능하게 합니다.
aug 목록은 아래 링크에서 확인가능합니다. https://albumentations.ai/docs/getting_started/transforms_and_targets/
aug 는 아래 링크에서 테스트 해볼 수 있습니다. https://yoyoyo-yo-albumentations-sim-sl-app-qevcnn.streamlit.app/
How to use
dataset config 내 train_pipeline 에 다음과 같은 코드를 넣어주면 됩니다.
예시
버그 발견시 알려주세요