[feat] Augmentation setup - Githubissues

boostcampaitech3 / level2-semantic-segmentation-level2-cv-17

[2022.04.25 ~ 2022.05.12] Recycle Trash Semantic Segmentation Competition - 부스트캠프 AI Tech 3기

4 stars 2 forks source link

[feat] Augmentation setup #12

Closed baekkr95 closed 2 years ago

baekkr95 commented 2 years ago

What is this PR?

10
Augmentation을 dataset 코드와 분리할 필요가 있다고 느껴짐

Changes

transform.py : 각자 원하는 augmentation 조합을 만들 수 있는 공간
dataset.py : transform.py에 있는 함수를 불러서 augmentation을 적용할 수 있음.

To reviewers

동우님이 현재 작업하시고 계신 파일 중에 train.py도 포함되어 있으신데, 제가 올린 PR에도 약간의 변화가 있습니다. 그 부분만 수정하시면 conflict는 없을 것 같습니다.

seonahmin commented 2 years ago

동우님이 수정해서 올려주신 PR과 차이가 있어서 헷갈리는 부분이 있는데 혹시 train.py에서도 수정되었다는 부분이 val_iou_by_cls = [0] * 11 이부분과 입력인 image랑 mask를 torch.stack한 그부분일까요?

baekkr95 commented 2 years ago

image와 mask을 stack한 부분입니당

Dongwoo-Im commented 2 years ago

@baekkr95 만약 transform.py를 github 추적없이 자유롭게 사용하는게 목적이라면, github에 파일을 올리지 않고 여기에 댓글로 적어주시는게 나을 것 같습니다!