datumaro.plugins.data_formats.imagenet

Classes

ImagenetBase(path, *[, subset, ctx])
ImagenetExporter(extractor, save_dir, *[, ...])
ImagenetImporter()	TorchVision's ImageFolder style importer.
ImagenetPath()
ImagenetWithSubsetDirsExporter(extractor, ...)
ImagenetWithSubsetDirsImporter()	TorchVision ImageFolder style importer.

class datumaro.plugins.data_formats.imagenet.ImagenetPath

Bases: object

IMAGE_DIR_NO_LABEL = 'no_label'

SEP_TOKEN = ':'

class datumaro.plugins.data_formats.imagenet.ImagenetBase(path: str, *, subset: str | None = None, ctx: ImportContext | None = None)

Bases: SubsetBase

class datumaro.plugins.data_formats.imagenet.ImagenetImporter

Bases: Importer

TorchVision’s ImageFolder style importer. For example, it imports the following directory structure.

root
├── label_0
│   ├── label_0_1.jpg
│   └── label_0_2.jpg
└── label_1
    └── label_1_1.jpg

classmethod detect(context: FormatDetectionContext) → FormatDetectionConfidence

classmethod find_sources(path)

classmethod get_file_extensions() → List[str]

class datumaro.plugins.data_formats.imagenet.ImagenetWithSubsetDirsImporter

Bases: ImagenetWithSubsetDirsImporter

TorchVision ImageFolder style importer. For example, it imports the following directory structure.

root
├── train
│   ├── label_0
│   │   ├── label_0_1.jpg
│   │   └── label_0_2.jpg
│   └── label_1
│       └── label_1_1.jpg
├── val
│   ├── label_0
│   │   ├── label_0_1.jpg
│   │   └── label_0_2.jpg
│   └── label_1
│       └── label_1_1.jpg
└── test
    ├── label_0
    │   ├── label_0_1.jpg
    │   └── label_0_2.jpg
    └── label_1
        └── label_1_1.jpg

Then, it will have three subsets: train, val, and test and they have label_0 and label_1 labels.

NAME = 'imagenet_with_subset_dirs'

classmethod detect(context: FormatDetectionContext) → FormatDetectionConfidence | None

class datumaro.plugins.data_formats.imagenet.ImagenetExporter(extractor: IDataset, save_dir: str, *, save_media: bool | None = None, image_ext: str | None = None, default_image_ext: str | None = None, save_dataset_meta: bool = False, save_hashkey_meta: bool = False, stream: bool = False, ctx: ExportContext | None = None)

Bases: Exporter

DEFAULT_IMAGE_EXT = '.jpg'

USE_SUBSET_DIRS = False

class datumaro.plugins.data_formats.imagenet.ImagenetWithSubsetDirsExporter(extractor: IDataset, save_dir: str, *, save_media: bool | None = None, image_ext: str | None = None, default_image_ext: str | None = None, save_dataset_meta: bool = False, save_hashkey_meta: bool = False, stream: bool = False, ctx: ExportContext | None = None)

Bases: ImagenetExporter

USE_SUBSET_DIRS = True