수화 이미지 분류 AI 해커톤

Computer Vision | Accuracy

상금 : 참가시 최소 50 XP, 특별상 데이콘 후드
473명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

우리는 꿈나무, Private 1위, Private 점수 :1, Noisy EfficientNet-v6

우리는꿈나무

2022.05.28 20:18 1,869 조회 language

안녕하세요. 운 좋게 1등을 하게 된 우리는 꿈나무 입니다.

저는 A-100 (40GB) GPU를 사용하여 Noisy EfficientNet-v6라는 모델을 학습 시켰습니다.
우선, 학습 전에 data의 pixel값 분포가 중요하다고 생각하여 train dataset의 mean, std 값을 구하여 해당 값으로 pixel값을 normalize를 하였습니다.
그리고 데이터의 양이 적다고 생각하여 augmentation된 데이터를 4배 추가 하였습니다.

제출할 때 기본 모델, voting, TTA(x3) 순으로 해보았는데 운 좋게 기본 모델에서 public score 0.99를 달성하여 해당 모델을 마지막으로 대회를 마무리하였습니다.

코드

댓글 7개

로그인이 필요합니다

comment

0 / 1000

bluestruggle

2022.05.30 08:54

코드 감사합니다. 제가 새내기라서 그런지 Noisy EfficientNet-v6라는 모델은 좀 생소한데, 관련 자료 링크가 있으면 부탁드려도 될까요?

우리는꿈나무

2022.05.31 14:51

https://arxiv.org/pdf/1911.04252.pdf 해당 모델의 논문 링크 드립니다

이치치

2022.05.31 10:41

안녕하세요, 저는 코드가 안 돌아가는데 확인해주실 수 있나요?

우리는꿈나무

2022.05.31 14:52

죄송합니다. 우선 지금 다시 보니 모델 저장하는 부분에서 vali_loss / len(vali_loader)로 수정해주어야 했는데 수정되지 않은 노트북을 올렸네요.
혹시 이 부분이 아니라 다른 부분에서 오류가 발생한다면 어느 부분에서 코드가 동작하지 않을까요?

소피아졔

2022.05.31 15:56

안녕하세요! train_transform 부분과 train dataset 정의 부분을 meanstd_loader 보다 앞쪽에 옮기면 train dataset is not defined이란 에러는 더이상 안뜨게 되네요 좋은 코드 감사드립니다.

우리는꿈나무

2022.05.31 17:56

추가적인 의견 드리자면, 해당 코드는 data augmentation을 하지 않은 상태에서 mean, std를 구하고 사용하기 위해 meanstd_loader를 따로 다른 cell에서 선언 후 해당 cell을 삭제 해주었습니다. 혹시나 사용하시게 된다면 data augmentation을 하지 않은 상태에서 진행해주셔야 합니다.

bluestruggle

2022.06.01 23:53

안녕하세요. train()에서 에러가 발생합니다.
print(train_loader)
---> 19 for img, label in train_loader:
20 print(img, label)
IndexError: list index out of range

분명 train_loader는 제대로 나욌거든요.
['./train/001.png', './train/002.png', ..
54912 54912
print(len(train_loader), len(valid_loader)) # 2746 687
무엇이 문제일까요?

📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동하였습니다!

목록으로

private 0.9074 ) tensorflow를 활용한 간단한 여러모듈 구현 및 ensemble

대회 - 수화 이미지 분류 AI 해커톤

2년 전

현재 글

우리는 꿈나무, Private 1위, Private 점수 :1, Noisy EfficientNet-v6

대회 - 수화 이미지 분류 AI 해커톤

2년 전

[Baseline] 1. Pytorch를 활용한 수화 이미지 분류

대회 - 수화 이미지 분류 AI 해커톤

2년 전