이미지다루기및 데이터셋 구축
[Dataset 구현] 1. customdataset
코딩펭귄
2023. 1. 5. 03:17
data를 가져오기위해 정의하는것 = dataset
init = 데이터의 path경로 정의
getitem : 이미지 하나하나 뽑아내서 어떻게 처리할지 정의 - 'transforms' 불러옴, 마지막엔 toTensor()처리해줘야됨(augmentation 끝나고 마지막엔 텐서처리해줘야됨)
customdata의 뼈대
- __init__()함수 : csv or 이미지 읽기, 변환할당, 데이터 필터링등 초기논리발생하는곳 (=초기화 해주는 곳이다.)
- __getitem__() : 데이터와 레이블을 반환, dataloader에서 호출됨, 단일데이터포인트(**텐서**, 넘파이 배열)에 대한 특정유형을 반환 ==> augmentation을 먹인 후 마지막에는 텐서형태를 띄어야 한다.(ToTensor() 해주기)
- __len() : 보유한 샘플수 반환(즉, 전체데이터 반환한다)''
customdata, dataset 만들기
cat_dog custom dataset 만들기
