def __init__(self, root, train=True, test=True, transform=None): # Data loading code std_value = 1.0 / 255.0 mean_values = [104 / 255.0, 117 / 255.0, 128 / 255.0] if transform is None: transform = [ transforms.Compose([ transforms.CovertBGR(), transforms.Resize(256), transforms.RandomResizedCrop(227), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize(mean=mean_values, std=3 * [std_value]), ]), transforms.Compose([ transforms.CovertBGR(), transforms.Resize(256), transforms.CenterCrop(227), transforms.ToTensor(), transforms.Normalize(mean=mean_values, std=3 * [std_value]), ]) ] if root is None: root = 'DataSet/Products' traindir = os.path.join(root, 'train') testdir = os.path.join(root, 'test') if train: self.train = datasets.ImageFolder(traindir, transform[0]) if test: self.test = datasets.ImageFolder(testdir, transform[1])