랜드마크 분류 AI 경진대회

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

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

연습

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

제출

파이토치를 이용하여 정말 간단하게 ResNet-101 학습부터 제출까지

우주대마왕

2020.11.06 12:05 21,660 조회 language

개발환경은 Ubuntu 20.04, 타이탄 RTX 2대, 라이젠 3950, 128GB, Python 3.7, PyTorch 1.6입니다.
정말 간단하게만 작성했습니다. 대회용 코드는 이것과 다릅니다.
단일 ResNet-101 모형에 이미지 크기 224, 배치 사이즈 64로 190,000번 학습하시면 스코어 0.95 이상 나옵니다.

모형을 EfficientNet으로 변경하고 ArcFace와 모형 학습 전략을 세우고 어그멘테이션을 albumentations,
스케줄러, 옵티마이저, AMP 등과 같이 다양한 방법을 시도하시면 더 좋은 성적을 달성하실 수 있을 것 같습니다.

코드