mmclassification训练注意事项

1 制作数据集

mmclassification的数据集是封装好的是ImageNet格式的。

所以首先需要把数据集先按照类别分别存放于一个子文件夹,所有子文件夹放在同一个大文件夹下。接着,通过随机划分训练集、验证集和测试集的代码,将三者划分成为8:1:1。然后制作标签。

2 模型

在configs文件中修改模型参数。明白模型配置,数据集配置和训练配置。最主要的了解一些主干模型的原理以及实现方式。

3 模型测试

模型路径与配置文件路径配置清楚然后使用tools/test.py对模型进行测试。

其中–metrics:评价方式,这依赖于数据集,比如准确率a

–metrics-options:对于评估过程的自定义操作,如topk=1

你可能感兴趣的:(python,人工智能,深度学习)