[PyTorch] Custom Dataset 만들기

모델을 만들면 항상 커스텀 데이터셋을 만들게 된다. 왜냐면 나는 이미지 폴더로 데이터로드 하는 것을 싫어한다. 아래와 같이 사용하는 것인데 이미지 폴더를 바로 가져와서 데이터셋을 구성하는 편리함이 있긴 하지만 데이터셋 내에서 데이터를 다양하게 조합하고 싶을 때마다 데이터를 매번 새로운 폴더로 만들 수는 없다. from torchvision.datasets import ImageFolder from torch.utils.data import DataLoader dataset = ImageFolder(root='folder_path', transform=transforms.Compose([ transforms.ToTensor(), ])) data_loader = DataLoader(dataset, batch..

2023.02.18