mmcls.models¶
The models
package contains several sub-packages for addressing the different components of a model.
classifiers
: The top-level module which defines the whole process of a classification model.backbones
: Usually a feature extraction network, e.g., ResNet, MobileNet.necks
: The component between backbones and heads, e.g., GlobalAveragePooling.heads
: The component for specific tasks. In MMClassification, we provides heads for classification.losses
: Loss functions.utils
: Some helper functions and common components used in various networks.data_preprocessor
: The component before model to preprocess the inputs, e.g., ClsDataPreprocessor.Common Components: Common components used in various networks.
Helper Functions: Helper functions.
Build Functions¶
Classifiers¶
Backbones¶
Necks¶
Heads¶
Losses¶
models.utils¶
This package includes some helper functions and common components used in various networks.