카메라 이미지 품질 향상 AI 경진대회

알고리즘 | CV | 이미지 변환 | 제어 | PSNR

  • moneyIcon 상금 : 총 1,000만원
  • 1,285명 마감

 

Test Time Augmentation

2021.07.27 19:52 4,065 조회 language

Baseline 코드에서 예측을 수행하는 Predict 함수의 코드를 기반으로 일반적인 Task에서 추가적인 성능 향상을 얻을 수 있는 Test Time Augmentation (TTA)을 구현해보았습니다. 이미지의 값을 바꿔주는 ColorJitter, Blur, Noise와 같은 Augmentation과 다르게 horizontal, Vertical Flip과 같은 이미지의 회전이 들어간 Augmentation의 경우는 TTA를 할 때, 회전 시켜서 예측한 후 다시 원래대로 회전시킨 후 합쳐 주어야 합니다.
아래와 같이 간단하게 구현할 수 있으므로 유용하게 사용하시면 좋을 것 같습니다. 다만, 이번 대회에서는 저희를 비롯한 다른 팀들도 TTA를 사용했을 때 약간의 성능 하락이 있다고 합니다. 잘 고려해서 사용하시면 좋을 듯 합니다.

코드