월간 데이콘 제 2회 컴퓨터 비전 학습 경진대회

저는 TF Keras 로 계속 코드를 작성해오다가 최근 torch를 통해 모델 구성을 하고 있습니다.

그래서 일부 적절하지 않은 코드가 있을 수 있습니다. 이런 부분은 피드백 부탁드립니다. 아직 torch 가 어색하네요 ㅠㅠ



=====================================================================================
-코드 설명-

데이터 로더를 통해 image를 3 channel로 메로리에 로드한 후 사용합니다.

transforms를 통해 random flip을 사용하였습니다.

모델은 EfficientNet-B0 pretrained imagenet을 사용하였고 output만 해당 대회에 맞게 수정하였습니다.

해당 코드에서는 데이터들을 5개의 fold로 나누지만 1fold만 실행을 합니다.

5fold를 실행하기 위해서는 for문에서 5번 반복하도록 수정하시면 됩니다.

2070SUPER에서 30epoch, 47 batch size 로 훈련 시키면 1fold당 약 3시간 소요 됩니다.

validation 성능이 개선될 때마다 가중치 저장을 해주었습니다.

훈련된 모델을 로드한 후 test 데이터를 추론하고 제출했을 때  accuracy 0.7869 점을 달성했습니다.
=====================================================================================

5fold, epoch 늘리기, TTA를 적용하면 성능이 80.xxx~81.xxx 정도 나옵니다.