图像分类降低损失,提高精度的方法(接前文章)

在demo测试时发现很多测试图片与标签不对应,准确率不是很高

在运行test后也可见下面精度并不高,打印的带标签的图片很多图片与标签不对应

图像分类降低损失,提高精度的方法(接前文章)_第1张图片  

针对这些情况,为降低损失,提高精度,我们可以为添加预训练模型进行训练

首先打开:

GitHub - open-mmlab-mmclassification: OpenMMLab Image Classification Toolbox and Benchmark

图像分类降低损失,提高精度的方法(接前文章)_第2张图片我们用的是resnet18_8xb32_in1k.py,故找到resnet,将imagenet-1k下的18层的模型下载到data文件下并解压

图像分类降低损失,提高精度的方法(接前文章)_第3张图片

 打开pycharm,复制data下刚刚存入的预训练模型的路径,打开咱们自己建的模型,在load_from下更改成自己的路径,再运行train.py训练,明确观察到损失减少的更快

图像分类降低损失,提高精度的方法(接前文章)_第4张图片

 运行test(删除注释符)

#../configs/resnet/fruit30_resnet18_8xb32_in1k.py
#../tools/work_dirs/fruit30/test1/epoch_100.pth
#--show-dir ../tools/work_dirs/fruit30/test1/result(不保存图片可删除本行)
#--metrics
#accuracy
#recall

可看到训练结果精度比原来不使用预训练模型明显提高,如重新打印了图片会发现图片的得分更高了,其准确率也大大提高了

图像分类降低损失,提高精度的方法(接前文章)_第5张图片

 

 

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