[Experiment] torch inference에 TTA 추가

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

[Experiment] torch inference에 TTA 추가 #19

Closed baekkr95 closed 2 years ago

baekkr95 commented 2 years ago

Background

inference.py에 tta를 추가하는 코드를 미리 세팅하려고 합니다.
어느정도의 성능 향상을 가져오는지 확인해보기 위해 실험해보려고 합니다.

Content

base 모델 : efficient b4 + Unet plusplus
tta 모델 : Horizontal, Rotate90, Multiply

Details

tta_transform = tta.Compose(
        [
            tta.HorizontalFlip(),
            tta.Rotate90([0, 90]),
            tta.Multiply([0.8, 1.0, 1.2])
        ]
    )

baekkr95 commented 2 years ago

pytorch tta reference : https://github.com/qubvel/ttach
pip install ttach 필요
한줄 평 : 일단 성능 자체는 올라간다...
추후에 argparser로 컨트롤할 수 있게 코드 추가할 예정입니다.

Dongwoo-Im commented 2 years ago

aux_params와 무관하게, albumentations 기반으로 TTA를 진행할 수 있는 코드를 작성 후 돌려보고 있습니다. 저번에 올려놨던 feat/torch-confusion branch에 push 후 댓글 남기겠습니다.