A PyTorch Dataset class that can be created from a folder `path` of images, for the sole purpose of inference. Optional `transforms`

FolderDataset(path, transforms = NULL)

Arguments

path

path to dir

transforms

transformations

Value

None

Details

can be provided. Attributes: `self.files`: A list of the filenames in the folder. `self.totensor`: `torchvision.transforms.ToTensor` transform. `self.transform`: The transforms passed in as `transforms` to the constructor.