MMClassifiction 框架学习导言
# MMClassifiction 框架学习导言
# 1 前言
因为想要搭一个自己分类以及分割的code base,想要先学习一下优秀的框架设计理念,所以先从较简单的图像分类看起,看一下 MMClassification 的设计架构以及代码风格,为之后的学术和竞赛搬砖搭一个顺手的框架
# 2 学习路径
第一阶段
- 首先了解是如何写config以及通过名称即能调用的(应该是注册)
- 了解 mmcls.dataset 是如何加载数据的
- 了解 mmcls.models 是如何搭建模型的
- 了解 logger 以及 tensorboard 等feature
- 目标:能够用类似的 config 形式调用模型,能够做一个自定义数据集的图像分类任务,并且有日志
第二阶段
支持多标签分类任务
添加 feature,例如 fp16、NNI、CAM等
了解 tests,完善测试流程
添加 tools
目标:能够有一些实用的小工具,能够 fp16 训练等
第三阶段
- 和落地部署结合起来
- 目标:在树莓派上部署ONNX、TensorRT等
上次更新: 2021/08/02, 21:04:52
- 02
- README 美化05-20
- 03
- 常见 Tricks 代码片段05-12