랜드마크 분류 AI 경진대회

이전 코드 공유에서 아주 간단한 모델로도 좋은 성능이 나오는 것을 확인하여 pretrained된 모바일넷 기반으로 학습을 진행해 본 예제입니다.
데이터는 이전에 제가 올린 게시글로 만든 TFRecords 기반의 데이터를 사용하였습니다.
컨볼루션 커널 단은 freezing하여 학습이 되지 않게 하였고, Global Average Pooling 이후인 classifier단만 학습이 되게 하였습니다.
이 것만으로도 training set에 완전히 fitting이 되는 것을 보아 모델의 capacity를 조금 더 낮춰도 될 듯 합니다만, 그런 모델은 pretrained weight를 찾기가 어렵네요 ㅠㅠ
성능은 Public score 기준 0.95038을 기록하였고, validation score와 크게 차이가 나지는 않는다는 점에서 테스트셋의 데이터가 학습 데이터와 크게 차이가 나지는 않는 것으로 보입니다 ㅎㅎ
코드 상단 모델 정의하는 부분에서 쉽게 다른 모델도 테스트 해보실 수 있습니다!