GitHub
Colab 教程
用命令行工具训练和推理
 用 Python API 训练和推理
Version
MMClassification 0.x
master branch
MMClassification 1.x
1.x branch
文档
MMEngine
MMCV
MMEval
MIM
MMAction2
MMClassification
MMDetection
MMDetection3D
MMEditing
MMGeneration
MMOCR
MMPose
MMSegmentation
MMTracking
MMFlow
MMFewShot
MMHuman3D
MMSelfSup
MMRazor
MMDeploy
MMRotate
MMYOLO
OpenMMLab
官网
 开放平台
 GitHub
推特
 知乎

Table of Contents

mmcls-1.x

开始你的第一步

依赖环境
安装

教程

使用现有模型推理
准备数据集
训练与测试
学习配置文件
如何微调模型

实用工具

数据集可视化
优化器参数策略可视化
类别激活图（CAM）可视化
打印完整配置文件
数据集验证
日志分析工具
模型复杂度分析
TorchServe 部署

进阶教程

添加新数据集
自定义数据处理流程（待更新）
自定义模型
自定义训练优化策略
自定义运行参数
自定义评估指标（待更新）
数据流（待更新）
MMCLS 中的约定

模型库

迁移指南

从 MMClassification 0.x 迁移

API 参考文档

mmcls.apis
mmcls.engine
mmcls.datasets
数据处理
mmcls.models
mmcls.structures
mmcls.visualization
mmcls.evaluation
mmcls.utils

其他说明

参与贡献 OpenMMLab
基于 MMClassification 的项目列表（待更新）
Changelog
常见问题

切换语言

English
简体中文

Docs >
mmcls.models >
mmcls.models.build_head
Read in English

Shortcuts

mmcls.models.build_head¶

mmcls.models.build_head(cfg)[源代码]¶: Build head.

Next

© Copyright 2020, OpenMMLab. Revision 3ff80f50.

Built with Sphinx using a theme provided by Read the Docs.

mmcls.models.build_head

GitHub
Colab 教程

用命令行工具训练和推理
用 Python API 训练和推理

Version

MMClassification 0.x
MMClassification 1.x