class torch.utils.data.Dataset
def My_Dataset(Dataset):
def __init__(self,data):
self.data = data
def __len__(self):
return len(self.data)
def __getitem__(self,idx):
pass
自定义的数据集抽象类,至少包含以上内容,否则会报错。
_len_:提供了数据集大小;
_getitem_:支持整数索引。
torchvision.datasets.ImageFolder(root,[transform,target_transform])
一个用于处理图片的、通用的Dataset类。
其接受的数据以文件夹-png图片格式的文件的形式组织。
其展示如下:
该类包含了以下的成员变量:
- self.classes : 用list保存的类名;
['cat', 'dog']
- self.idx_to_class :类名对应的索引;
{'cat': 0, 'dog': 1}
- self.imgs :保存(img_path,class)的list。
[('./data/dogcat_2/cat/cat.12484.jpg', 0),
('./data/dogcat_2/cat/cat.12485.jpg', 0),
('./data/dogcat_2/cat/cat.12486.jpg', 0),
('./data/dogcat_2/cat/cat.12487.jpg', 0),
('./data/dogcat_2/dog/dog.12496.jpg', 1),
('./data/dogcat_2/dog/dog.12497.jpg', 1),
('./data/dogcat_2/dog/dog.12498.jpg', 1),
('./data/dogcat_2/dog/dog.12499.jpg', 1)]