YOLOv5源码逐行超详细注释与解读(4)——验证部分val(test).py

慈云数据 8个月前 (03-12) 技术支持 138 0

前言 

本篇文章主要是对YOLOv5项目的验证部分。这个文件之前是叫test.py,后来改为val.py。

在之前我们已经学习了推理部分detect.py和训练部分train.py这两个,而我们今天要介绍的验证部分val.py这个文件主要是train.py每一轮训练结束后,用val.py去验证当前模型的mAP、混淆矩阵等指标以及各个超参数是否是最佳, 不是最佳的话修改train.py里面的结构;确定是最佳了再用detect.py去泛化使用

总结一下这三个文件的区别

微信扫一扫加客服

微信扫一扫加客服

点击启动AI问答
Draggable Icon