Torchvision Dataset

Classes:

TorchDataset(torch_dataset)

Wrapper class to handle datasets loaded from torchvision.

class cleanvision.dataset.torch_dataset.TorchDataset(torch_dataset)

Bases: Dataset

Wrapper class to handle datasets loaded from torchvision.

Methods:

get_name(index)

Returns the name of the image in the dataset.

get_name(index)

Returns the name of the image in the dataset. It can be a filename or a str with index information.

Return type:

str