dataset_SVHN.py 文件源码

def __init__(self, batchsize, train=True):
        Dataset.__init__(self)
        data_root = join(dirname(realpath(__file__)), 'SVHN_data')
        self.name = "svhn"
        self.range = [0.0, 1.0]
        self.data_dims = [3, 32, 32]
        self.batchsize = batchsize
        if train:
            split = "train"
        self.data = dsets.SVHN(root=data_root,
                           download=True,
                           split="train",
                           transform=transforms.Compose([
                                transforms.ToTensor()]))
        self.dataloder = tdata.DataLoader(self.data, self.batchsize, shuffle=True)
        self.iter = iter(self.dataloder)
        self._index = 0