Shortcuts

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.

Build Functions

Classifiers

Backbones

Necks

Heads

Losses

models.utils

This package includes some helper functions and common components used in various networks.

Common Components

Helper Functions