torch_concepts.data.datasets.awa2.AwA2Dataset

class AwA2Dataset(root, training_augment=True, split='train', image_size=224, concept_transform=None, sample_transform=None, selected_concepts=None, seed=42)

Returns a compatible Torch Dataset object customized for the AwA2 dataset

__init__(root, training_augment=True, split='train', image_size=224, concept_transform=None, sample_transform=None, selected_concepts=None, seed=42)

Methods

__init__(root[, training_augment, split, ...])