한 번만 더 해보자

[Deep Learning] Test Time Augmentation (TTA) 본문

Deep Learning

[Deep Learning] Test Time Augmentation (TTA)

정 하임 2024. 2. 1. 21:41

Augmentation

  • 딥러닝 모델을 학습시킬 때 부족한 데이터를 늘리기 위해서 임의로 변형을 주는 Augmentation 사용함
  • crop, flip, blur 등을 사용해서 이미지에 변형을 준다

TTA란?

  • Test Time Augmentation
  • Augmentation을 추론 과정에서도 사용하는 것
  • 학습이 완료된 모델을 평가할 때, Augmentation을 통해 2장의 이미지를 추가하고 전부 Inference를 수행한다 그리고 그 결과물의 평균을 내서 class를 할당한다
  • 원본으로부터 변형된 여러가지 Image Augmentation에 평가를 실시하여 최종 분류값이 무엇인지 예측하는 기법

반응형