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)
path to dir
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.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.