コード例 #1
0
ファイル: fmnist.py プロジェクト: zhwzhong/vega
 def __init__(self, **kwargs):
     """Construct the Imagenet class."""
     Dataset.__init__(self, **kwargs)
     FashionMNIST.__init__(self,
                           root=self.args.data_path,
                           train=self.train,
                           transform=Compose(self.transforms.__transform__),
                           download=self.args.download)
コード例 #2
0
ファイル: fmnist.py プロジェクト: vineetrao25/vega
 def __init__(self, **kwargs):
     """Construct the Imagenet class."""
     Dataset.__init__(self, **kwargs)
     self.args.data_path = FileOps.download_dataset(self.args.data_path)
     FashionMNIST.__init__(self, root=self.args.data_path, train=self.train,
                           transform=self.transforms, download=self.args.download)