mmagic.datasets.imagenet_dataset

Module Contents

Classes

ImageNet

ImageNet Dataset.

class mmagic.datasets.imagenet_dataset.ImageNet(ann_file: str = '', metainfo: Optional[dict] = None, data_root: str = '', data_prefix: Union[str, dict] = '', **kwargs)

Bases: mmagic.datasets.basic_conditional_dataset.BasicConditionalDataset

ImageNet Dataset.

The dataset supports two kinds of annotation format. More details can be found in CustomDataset.

Parameters

• ann_file (str) – Annotation file path. Defaults to ‘’.

• metainfo (dict, optional) – Meta information for dataset, such as class information. Defaults to None.

• data_root (str) – The root directory for data_prefix and ann_file. Defaults to ‘’.

• data_prefix (str | dict) – Prefix for training data. Defaults to ‘’.

• **kwargs – Other keyword arguments in CustomDataset and BaseDataset.

IMG_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif')

METAINFO