랜드마크 분류 AI 경진대회

알고리즘 | 이미지 | 분류 | 상권 | GAP

상금 : 1,500만원
678명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

Google colab으로 TPU이용해서 학습하기

DDaan

2020.11.06 16:57 8,002 조회 language

학습데이터는 Google Cloud에 업로드 하여 진행했습니다. Google Cloud를 처음시작하면 $300 무료 기프티콘을 3개월동안 사용할 수 있으니 충분히 무료로 즐기실 수 있으세요! Cloud storage같은 경우는 요금 지출이 많지 않아서 ㅎㅎ

Tfrec로 데이터를 변환시켜 학습을 진행했구요. 학습모델은 EfficientNetB7을 예시로 들었습니다. 학습시간이 오래걸려 끊어지면 계속 연결하고 이어서 학습해줘야 하므로 빠른 결과를 보고 싶으신 분은 EfficientNetB0~5 를 사용해서 학습시켜보세요!
EfficientNetB7 50epochs 결과 0.967정도가 나왔습니다.
시간이 있으면 여러 모델을 학습하고 ensemble등을 진행해보고 싶지만 남은 기한이 별로 없네요.

코드

댓글 2개

로그인이 필요합니다

comment

0 / 1000

DDaan

2020.11.06 17:46

tfrec 변환하는 코드
https://drive.google.com/file/d/1F6LvbEjkbJn8jVLqONU78QT4dqG13SpB/view?usp=sharing
이고, 이 작업은 로컬피시로 작업 후, tfrec파일들을 google cloud에 업로드 하였습니다.
train_tfrec/
   train-00000-of-00032-2215.tfrec
   train-00001-of-00032-2215.tfrec
   ...