发布时间:2023-03-31 文章分类:电脑基础 投稿人:樱花 字号: 默认 | | 超大 打印

模型评价指标

  • 检测精度
  • 检测速度

本文训练了50轮安全帽检测数据集,分析训练结果如下:

检测精度 检测速度
Precision;Recall;F1_score 前传耗时
IOU交并比 FPS每秒帧数
P-R曲线 flops浮点运算数量
AP;mAP

检测精度

  1. 混淆矩阵confusion_matrix

yolov5模型训练结果分析

  1. P&R&P-R&F1_curve
  • 准确率Precision:(检测的准不准确)检测的效果好不好,检测到的区域是不是都为正确区域。P=TP/TP+FP
  • 理想值:1
  • 召回率Recall:(检测的全不全面)该检测到的区域是不是都检测到了。R=TP/TP+FN;FN是该检测到但是没有检测出的区域。
  • 理想值:1

该图描述随着置信度阈值的增加,P值的变化;置信度设为某一数值的时候,各个类别识别的准确率。

当设置置信度为某一数值的时候,各个类别的查全的概率。

精度和召回率的关系图,而是是有些矛盾存在的。精度越高,召回率越低,理想情况是(1,1)点,即在准确度很高的情况下,尽可能检测到全部的类别。二者围成的面积就是mAP值,mAP面积越接近于1,效果越好。

yolov5模型训练结果分析
F1-score是分类问题的一个衡量指标。是精确率和召回率的调和平均数,最大为1,最小为0。 F1_score=2*P*R/P+R

  1. Labels:

yolov5模型训练结果分析

第一个图是训练集的数据量,每个类别有多少个
第二个是框的尺寸和数量
第三个是center点的位置。
第四个是labeld的高宽。可以看出头部一般相比于整个图片比较小,所以看到样本大多分布在(0-0.2,0-0.2)
4. result.png

yolov5模型训练结果分析

检测速度

参考:
YOLO-V5训练结果的分析与评价:https://blog.csdn.net/weixin_45751396/article/details/126726120