Hello Everyone,
I had a question about Torchvision Datatsets. Are these datasets its own class or is it just a link where you can download the files? I’m trying run SRGAN sample program and it uses torchvision dataset but I wanted to run my own dataset but getting errors. Any help would be greatly appreciated! Thank you!
USING_STL = True
if USING_STL:
DatasetSubclass = torchvision.datasets.STL10
else:
DatasetSubclass = torchvision.datasets.ImageNet
class Dataset(DatasetSubclass):
def init(self, *args, **kwargs):
hr_size = kwargs.pop(‘hr_size’, [96, 96])
lr_size = kwargs.pop(‘lr_size’, [24, 24])
super().init(*args, **kwargs)
How do I replace STL10 or ImageNet dataset with my own dataset?