Description

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

Usage

Arguments

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