대학 대항전 : 퍼즐 이미지 AI 경진대회

먼저 코드 공유를 해주신 파이썬초보만님께 감사합니다.
jigsawformer에 변형을 주었습니다.
크게 2가지를 추가하고, hyper-parameter를 조정하였습니다.
1. Data augmentation (다양한 data augmentation을 통해서 모델의 강건성을 확보하고자 하였습니다)
- 특히 cutout 계열을 사용했을 때, 보다 효과적이었습니다.
2. flip / rotate 등 다양한 조건 부여
- 코드 진행 방식은 크게 다음과 같습니다
a) original image (순서가 섞인 이미지)를 입력으로 받아 정답 label을 통해 초기화
b) (핵심) 초기화된 이미지와 label에 flip / rotate를 확률적으로 가함
→ 단순히 제대로 된 방향의 퍼즐뿐만 아니라 뒤집히거나 돌아간 퍼즐 또한 잘 풀어야 함
c) 그 상태에서 random으로 섞어줌
pyper-parameter는 다음과 같습니다
1. optimizer AdamW 
2. scheduler : CosineAnnealingWarmRestart
3. Batchsize 224 for train / 224 * 8 for validation / 224 * 8 for test
4. GPU : RTX A6000 48GB x 1EA