YOLOv5在验证集上进行测试

前言

今天想在验证集上测试模型的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文件