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

알고리즘 | 월간 데이콘 | 대학 대항전 | 비전 | 자기지도학습 | 기술 혁신 | 분류

상금 : 인증서
842명 마감

연습

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

제출

[Private 4th] Solution

_Quokka_

2024.02.05 21:01 359 조회 language

jigsawformer 코드를 변형하여 사용하였습니다.
파이썬초보만님 좋은 코드 너무 감사드립니다.

1. Optimizer 변경 + LR scheduler 추가
기존 Adam에서 AdamW와 CosineAnnealingLR 조합을 사용

2. 다양한 Data Augmentation 적용
- Random Photometric Distort
- Random Erasing
- Random Vertical Flip
- Random Horizontal Flip
- Random Rotate 90, 180, 270
- Random Cutout (Before Shuffle)
- Test Time Augment (Flip)

3. 여러개의 Pred 값들 Voting Ensemble 진행

학습 환경
- Ryzen 5600x
- Ram 32GB
- RTX 3090 24GB
- Windows WSL

Batch Size == 64 (with AMP)

Python 3.8.18
torch.__version__ == 2.1.2
Cuda compilation tools, release 11.5