len
方法,提供了dataset
的大小; getitem
方法, 该方法支持从 0
到 len(self)
的索引。
#filenames是训练数据文件名称列表,labels是标签列表
class MyDataset(Dataset):
def __init__(self, filenames, labels, transform):
self.filenames = filenames
self.labels
len
方法,提供了dataset
的大小; getitem
方法, 该方法支持从 0
到 len(self)
的索引。
#filenames是训练数据文件名称列表,labels是标签列表
class MyDataset(Dataset):
def __init__(self, filenames, labels, transform):
self.filenames = filenames
self.labels