월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회

알고리즘 | CV | 분류 | 이상탐지 | Macro f1 score

Prize : 100만원 + ɑ
1,392명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

[PyTorch] 이미지를 레이블별로 저장하고 ImageFolder를 사용하여 data split까지 해보자

11kwak

2022.04.07 18:41 23,755 Views language

torch 사용자들에게는 ImageFolder 라는 아주 강력한 무기가 존재합니다.
폴더명을 레이블로하여 이미지만 분류할 수 있다면 data split하기에 너무 편한 환경을 구축할 수 있습니다.
다행히도 데이콘측에서 이미지명에 따른 레이블을 알려주었기 때문에 쉽게 데이터 전처리를 마칠 수 있었습니다.
mean/std에 따른 이미지 normaliztion까지 적용하여 train data/ valid data 로 7:3 비율 data split 까지 해보겠습니다.
이제 여러분은 이어서 데이터 불균형 문제를 해결하고, 모델을 구축하고, dataloader를 불러와 학습과 평가를 진행하시면 됩니다.

Code

Comments 7개

로그인이 필요합니다

comment

0 / 1000

dinggulman

2022.04.08 10:43

삭제된 댓글입니다

데이터분석잘하고싶은사람

2022.04.14 14:24

감사합니다!

victory

2022.04.21 16:04

삭제된 댓글입니다

steven

2022.05.01 22:02

삭제된 댓글입니다

ingbeeeded

2022.05.06 19:29

label = train_df.loc[train_df["file_name"] == f"{train_folder[i]}"]["label"][i]

이 부분

label = train_df.loc[train_df["file_name"] == f"{train_folder[i]}"]["label"].item()으로 수정하면 되던데 맞나요?

minkyoon

2023.04.21 16:37

우와 감사합니다. 잘 사용하겠습니다,,

KimVision

2023.04.26 02:42

label = train_df.loc[train_df["file_name"] == f"{train_folder[i]}"]["label"][i] # train_df에서 이미지 이름에 맞는 label을 불러와 저장

이 구간에서 index 에러가 발생 하는데, 혹시 해결 방법이 어떤 건지 알 수 있을까요??

위에분이 남기신 것처럼 .item() 함수 호출해도 다른 에러 발생하고 해결이 안되서여!

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

List

hydra + pytorch_lightning 기반 코드 공유

Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회

현재 글

[PyTorch] 이미지를 레이블별로 저장하고 ImageFolder를 사용하여 data split까지 해보자

Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회

Likes 36