이미지 분류 해커톤: 데이터 속 아이콘의 종류를 맞혀라!

데이콘 해커톤 | 알고리즘 | 입문 | 비전 | 분류 | Accuracy

  • moneyIcon 상금 : 데이스쿨 프로 구독권
  • 2025.03.04 ~ 2025.04.30 09:59 + Google Calendar
  • 742명 마감

 

Private 18위, Densenet201 모델, 이미지 Flip(상하좌우) 증강 적용

2025.05.02 17:59 178 조회 language

timm 라이브러리에 있는 densenet201 모델을 pretrained = False로 지정하여 사전학습 없이 아키텍처만 불러와 사용하였습니다.
학습데이터의 이미지를 상하좌우 뒤집기만 적용하여 학습 진행하였습니다.

# 사용 라이브러리 버전
# numpy : 2.2.4
# pandas : 2.2.3
# torch : 2.6.8+cu124
# torchvision : 0.21.0+cu124
# timm : 1.0.15
# scikit-learn : 1.6.1
# tqdm : 4.67.1

코드