FolderDataset {fastai}

FolderDataset

Description

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

Usage

FolderDataset(path, transforms = NULL)

Arguments

path

path to dir

transforms

transformations

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.

Value

None