一般的训练流程
- 数据集准备和可视化
- 自定义配置文件
- 训练前可视化验证
- 模型训练
- 模型测试和推理
- 可视化分析
官方演示文档:https://github.com/open-mmlab/mmdetection/blob/tutorials/demo/MMDet_Tutorial.ipynb
0 环境检测与安装
检查nvcc、gcc版本
安装openmim:mim是一个OpenMMLab的包管理器,它相当于pip。它会对OpenMMLab的算法库进行额外的管理。比如安装mmcv的时候,大家经常会不知道cuda版本是多少。一旦mim安装,它会自动基于你的底层环境,来选择最合适你的预编译包。它如果发现你的环境中没有预编译包,它可能会报错,因为你需要设置nvcc的版本。
啥是预编译包?
1 数据集准备和可视化
目标检测领域基本以CoCo标注格式为标准的。