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

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

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

 

Pytorch로 CNN 직접 구축하기 (Public 0.871)

2021.11.24 12:06 3,950 Views language

안녕하세요. 자몽허니블랙티입니다.

저는 데이콘에서 2020년에 개최됐던 월간 데이콘 '컴퓨터 비전 학습 경진대회'에 참여했습니다. (https://dacon.io/competitions/official/235626)
당시에 Tensorflow를 사용해서 CNN 모델을 만들었는데, 이번에는 Pytorch를 사용해서 모델을 구축해보았습니다. (구조는 동일합니다.)

같은 MNIST 기반이긴 하지만, 기존 대회와 데이터셋이 다르다보니 데이터셋에 맞게 튜닝을 하면 더 좋은 점수가 나올 것 같습니다.
저는 개인적으로 시간이 부족하여 여기까지만 진행하고 공유드립니다.
해당 코드 활용하여 점수가 더 개선되는 분들은 공유해주시면 감사하겠습니다.
출처만 밝혀주시면 코드를 자유롭게 사용하셔도 됩니다.

Code
로그인이 필요합니다
0 / 1000
DACON.tuna
2021.11.24 15:50

오우 torch 공부 중인데 감사합니다!