mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
* 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>
84 lines
1.2 KiB
ReStructuredText
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.
|