datasets.rst

.. role:: hidden
    :class: hidden-section

.. module:: mmpretrain.datasets

mmpretrain.datasets
===================================

The ``datasets`` package contains several usual datasets for image classification tasks and some dataset wrappers.

.. contents:: mmpretrain.datasets
   :depth: 2
   :local:
   :backlinks: top

Custom Dataset
--------------

.. autoclass:: CustomDataset

ImageNet
--------

.. autoclass:: ImageNet

.. autoclass:: ImageNet21k

CIFAR
-----

.. autoclass:: CIFAR10

.. autoclass:: CIFAR100

MNIST
-----

.. autoclass:: MNIST

.. autoclass:: FashionMNIST

VOC
---

.. autoclass:: VOC

CUB
---

.. autoclass:: CUB

Places205
---------

.. autoclass:: Places205

Retrieval
---------

.. autoclass:: InShop

Base classes
------------

.. autoclass:: BaseDataset

.. autoclass:: MultiLabelDataset

Caltech101
----------------

.. autoclass:: Caltech101

Food101
----------------

.. autoclass:: Food101

DTD
----------------

.. autoclass:: DTD

FGVCAircraft
----------------

.. autoclass:: FGVCAircraft


Flowers102
----------------

.. autoclass:: Flowers102

StanfordCars
----------------

.. autoclass:: StanfordCars

OxfordIIITPet
----------------

.. autoclass:: OxfordIIITPet

SUN397
----------------

.. autoclass:: SUN397

RefCOCO
--------

.. autoclass:: RefCOCO

Dataset Wrappers
----------------

.. autoclass:: KFoldDataset

The dataset wrappers in the MMEngine can be directly used in MMPreTrain.

.. list-table::

   * - :class:`~mmengine.dataset.ConcatDataset`
     - A wrapper of concatenated dataset.
   * - :class:`~mmengine.dataset.RepeatDataset`
     - A wrapper of repeated dataset.
   * - :class:`~mmengine.dataset.ClassBalancedDataset`
     - A wrapper of class balanced dataset.