pytorch 커스텀 데이터셋 만들기 Custom Dataset

저번 포스팅에서는 TensorDataset 클래스를 이용해서 텐서의 입력을 받아 Dataset을 정의하고, DataLoader 클래스를 이용해서 data shuffle과 mini-batch 학습을 수행하는 방법에 대해서 다뤄봤습니다. 즉, 데이터를 정의하고 데이터를 불러오는 방법에 대해서 배웠습니다. from torch.utils.data import TensorDataset from torch.utils.data import DataLoader 이번 포스팅에서는 torch.utils.data.Dataset을 상속 받아 직접 Custom Dataset을 만드는 방법에 대해서 알아보겠습니다. 즉, 이번에는 클래스 형태로 데이터를 정의하는겁니다. 이후에 DataLoader를 이용해서 데이터를 불러오는 과정은 동일합니다. torch.utils.data.Dataset 상속 받아서 Custom Dataset 만들기 본 포스팅에 들어가기 전에, 커스텀 데이터셋을 만들 때 가장 기본적인 뼈대는 ...

원문링크 : pytorch 커스텀 데이터셋 만들기 Custom Dataset