130 lines
1.7 KiB
ReStructuredText
130 lines
1.7 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
|
|
|
|
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.
|