mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
496e098b21
* feat: support some downstream classification datasets * update sun397 * sum * update sun397 * [CI] Add test mim CI. (#879) * feat: support some downstream classification datasets * update sun397 * sum * update sun397 * rebase * feat: support some downstream classification datasets * update sun397 * update sun397 * update sun397 * update sun397 * fix unittest * update docstring * rm * update * update * refactor names of datasets * refactor some implements of datasets * refactor some implements of datasets * fix datasets unittest * refactor cub and stanford cars * refactor cub and cifar * refactor cub and cifar * refactor cub and cifar * update downstream datasets and docs * update docstring --------- Co-authored-by: Ma Zerun <mzr1996@163.com> Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
125 lines
1.7 KiB
ReStructuredText
125 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
|
|
|
|
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.
|