손 글씨 (숫자) 분류 AI 해커톤

알고리즘 | CV | 분류 | 제어 | Accuracy

  • moneyIcon Prize : 참가시 최소 50 XP, 특별상 데이콘 후드
  • 450명 마감

 

PV 15등 Se-ResNet, (cutmix, morphology Aug)

2021.11.29 17:19 2,108 Views language

안녕하세요 코드 공유합니다
28x28임을 감안하면, torchvision이나 timm은 224x224에 맞춰졌기 때문에 적절하지가 않아서
깊지 않는 네트워크 중 Se-ResNet 오픈소스 네트워크를 이용해서 훈련을 했습니다
데이터가 많지 않아, 여러 Augmentation을 적용했는데 주요 Augmentation은 cutmix, morphology(dilation, erosion, 침식 팽창)을 이용했습니다
마지막에 tta를 섞으니 점수가 더 잘 나왔습니다
감사합니다

Code