IT

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

ShrimpTaco 2023. 4. 29. 16:10
반응형

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

 

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

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

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

 

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

 

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

 

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

 

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

 

EDA

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

 

학습

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

 

모델 개선

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

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

반응형