mmclassification/docs/en/api/datasets.rst
Yixiao Fang 08dc8c75d3
[Refactor] Add selfsup algorithms. (#1389)
* remove basehead

* add moco series

* add byol simclr simsiam

* add ut

* update configs

* add simsiam hook

* add and refactor beit

* update ut

* add cae

* update extract_feat

* refactor cae

* add mae

* refactor data preprocessor

* update heads

* add maskfeat

* add milan

* add simmim

* add mixmim

* fix lint

* fix ut

* fix lint

* add eva

* add densecl

* add barlowtwins

* add swav

* fix lint

* update readtherdocs rst

* update docs

* update

* Decrease UT memory usage

* Fix docstring

* update DALLEEncoder

* Update model docs

* refactor dalle encoder

* update docstring

* fix ut

* fix config error

* add val_cfg and test_cfg

* refactor clip generator

* fix lint

* pass check

* fix ut

* add lars

* update type of BEiT in configs

* Use MMEngine style momentum in EMA.

* apply mmpretrain solarize

---------

Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-06 16:53:15 +08:00

84 lines
1.2 KiB
ReStructuredText

.. 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
Dataset Wrappers
----------------
.. autoclass:: KFoldDataset
The dataset wrappers in the MMEngine can be directly used in MMClassification.
.. 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.