HuggingFace Dataset#

Classes:

HFDataset(hf_dataset, image_key)

Wrapper class to handle datasets loaded from Huggingface.

class cleanvision.dataset.hf_dataset.HFDataset(hf_dataset, image_key)[source]#

Bases: Dataset

Wrapper class to handle datasets loaded from Huggingface.

Methods:

get_name(item)

Returns the name of the image in the dataset.

get_name(item)[source]#

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

Return type:

str