qq_39756856的博客

原创 2021-06-24

使用mmclassification训练自己的数据集介绍本文环境准备数据集修改mmclassification代码修改配置文件训练测试生成onnx文件介绍前段时间正好在做一个项目的过程中需要使用mmclassification训练自己的数据集，顺便整理了一下流程供大家参考。本文章默认大家已经安装好mmcv的所有环境。本文环境cuda ->11.1torch ->1.9.0python ->3.6mmcv-full ->1.3.0mmcls ->0.12.0全