前言
今天想在验证集上测试模型的mAP,但翻遍了整个项目目录并没有找到val.py,这就很奇怪
经过在Github上yolov5的原项目中查找,发现作者在2021年7月14日已经将test.py更名为val.py
这样就已经确定是自己的yolov5的版本较低,仍采用的是test.py(v6.0的版本就更新为val.py了),但是在验证集上进行测试还是没有问题的
步骤
1.修改配置文件
打开test.py文件,下拉至最后的主函数部分,将模型权重文件以及数据集配置文件更换为自己的
2.运行
更改好之后就可以直接运行了,若不想修改配置文件,直接在命令行输入以下代码也是可以的,注意替换文件名即可
$ python test.py --weights yolov5s.pt --data coco128.yaml
3.查看结果
运行结束后的结果会保存至run/test/文件夹下,其中包括confusion_matrix(混淆矩阵)等文件,mAP信息也可以找到,这样就算大功告成了
总结
1.在yolov5-6.0的版本中,test.py就已经被重命名为val.py,作者解释其原因是为了避免将来与 python包发生任何冲突和歧义。文章来源:https://www.toymoban.com/news/detail-508690.html
2.推荐使用最新版的yolov5,使用起来更顺手一点,而且新版的yolov5以及可以导出更多的种类的模型进行使用,详情可看新版本中的export.py文件文章来源地址https://www.toymoban.com/news/detail-508690.html
到了这里,关于YOLOv5在验证集上进行测试的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!