Base Dataset#

Classes:

Dataset()

Wrapper class to handle datasets loaded from various sources like: image files in a local folder, huggingface, or torchvision.

class cleanvision.dataset.base_dataset.Dataset[source]#

Bases: Sized

Wrapper class to handle datasets loaded from various sources like: image files in a local folder, huggingface, or torchvision.

Methods:

get_name(index)

Returns the name of the image in the dataset.

get_name(index)[source]#

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

Return type:

str