前言

今天想在验证集上测试模型的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包发生任何冲突和歧义

2.推荐使用最新版的yolov5,使用起来更顺手一点,而且新版的yolov5以及可以导出更多的种类的模型进行使用,详情可看新版本中的export.py文件

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐