qq_39756856的博客

原创 2021-06-24

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