图像分类训练自己的任务及模型评估

目录

一.数据集的存放

 二.新建模型

 三.修改参数

 四.定义标签的两种方法

1.直接在imagenet修改

 2.新建文件并注册

 五.单张图片测试

 六.模型评估

---

一.数据集的存放

首先在mmcls目录下新建data文件用来存我们的数据集

将自己的数据集放入到data文件夹中，我这用的是一个水果的数据集，并且已经划分好了训练集和验证集

  打开训练集可以看到里面有很多种类

 二.新建模型

打开我们的pycharm工程，以resnet下的resnet18_8xb32_in1k.py为例

先找到configs路径下的resnet,复制resnet18_8xb32_in1k.py的绝对路径

打开train.py文件，将路径粘贴并改为双斜杠，注释掉：

# C:\\Users\\lenovo\\MMCLS1\\mmclassification-master\\configs\\resnet\\resnet18_8xb32_in1k.py

将此路径复制到train运行配置中的形参中，并运行train

这里遇到报错不用管，再打开work_dirs可看到里面生成了一个resnet18_8xb32_in1k.py文件夹，打开后里面有一个resnet18_8xb32_in1k.py文件，我们复制一下这个文件，再打开pycharm

 将刚刚复制的文件粘贴到resnet目录