본문 바로가기
IT

데이콘 일기 (1) - 이미지 분류 경진대회 준비하기

by ShrimpTaco 2023. 4. 29.
반응형

공부를 시작하며 데이콘 경진대회에 대해 알아보았다.

 

흥미로워 보이는 이미지 분류 경진대회가 진행 중이었다.

도배 하자 유형 분류 AI 경진대회

대회 기간은 4월 10일부터 5월 22일까지.

 

많은 참가자들이 결과물을 제출할 수 있도록 Pytorch로 작성된 EfficientNet B0 베이스라인 코드를 데이콘에서 제공한다.

 

나는 TensorFlow가 익숙할 것 같아서 제공된 Baseline을 사용하지 않을 거지만, 이미지 분류 모델로 EfficientNet이 유명하다는 것을 알게 되었다.

 

지난 대회들의 코드 공유 게시판을 보니 이전 대회에 참여하고 순위권에 들었던 사람들이 코드를 공유해 줘서 공부할 때 참고할 수 있을 것 같다.

 

우선 대강의 계획을 세워보면 다음과 같다.

 

EDA

  • 데이터 라벨 분포 확인
  • 일부 학습 데이터 plot 해서 확인

 

학습

  • EfficientNet 공부
  • TensorFlow로 EfficientNet 사용하는 법 공부

 

모델 개선

  • 데이터 증강
  • EfficientNet의 사이즈 변경 (B0 ~ B7)
  • 기타 요소 확인

성능이 좋으면 좋겠지만, 우선은 기간 내에 원하는 대로 동작하는 코드를 작성하고, 제출할 수 있는 결과물을 얻어내는 것까지를 목표로 한다.

반응형

댓글