1 制作数据集
mmclassification的数据集是封装好的是ImageNet格式的。
所以首先需要把数据集先按照类别分别存放于一个子文件夹,所有子文件夹放在同一个大文件夹下。接着,通过随机划分训练集、验证集和测试集的代码,将三者划分成为8:1:1。然后制作标签。
2 模型
在configs文件中修改模型参数。明白模型配置,数据集配置和训练配置。最主要的了解一些主干模型的原理以及实现方式。
3 模型测试
模型路径与配置文件路径配置清楚然后使用tools/test.py对模型进行测试。
其中–metrics:评价方式,这依赖于数据集,比如准确率a
–metrics-options:对于评估过程的自定义操作,如topk=1