VISION HONG
article thumbnail
Custom Dataset
Pytorch 2020. 12. 14. 21:10

Pytorch에서는 사용자가 추상클래스인 torch.utils.data.Dataset을 오버라이드하여 직접 커스텀 데이터셋을 만들 수 있도록 지원한다. 커스텀 데이터셋을 만들 때, 가장 기본적인 뼈대는 아래와 같다. (아래 3개의 메서드는 반드시 구현되어야함.) class CustomDataset(torch.utils.data.Dataset): def __init__(self): def __len__(self): def __getitem__(self, idx): __init__ : 데이터셋 전처리 __len__ : 데이터셋의 총 길이. 즉 총 데이터 수 __getitem__ : 어떤 샘플을 가져올지 인덱스를 받아서 그만큼 보내주는 함수 이제 실제 데이터를 가지고 커스텀 데이터셋을 만들어보자. 데이터는 유..