vision_unlearning.datasets.coco

Classes

UnlearnDatasetCoco

Wrapper around huggingface datasets

Module Contents

class vision_unlearning.datasets.coco.UnlearnDatasetCoco(/, **data: Any)[source]

Bases: vision_unlearning.datasets.base.UnlearnDataset

Wrapper around huggingface datasets Organize the forget-retain splits

abstract _load() None[source]

Load the dataset from disk or download it. Side effects: updates the properties _dataset_splits, _classes, _n_classes