You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

YOLOv4-tiny训练日志解读:损失与IOU选择及术语疑问

YOLOv4-tiny训练日志解析与绘图数值选择

一、日志关键术语说明

  • v3:YOLOv4-tiny沿用了YOLOv3的损失计算逻辑(包含IOU损失、目标置信度损失、类别损失的组合规则),因此日志中会标注v3。
  • Region 30、Region 37:对应YOLOv4-tiny的两个检测输出层:
    • Region 30是下采样16倍的大尺度检测层,负责检测图像中的小目标;
    • Region 37是下采样32倍的小尺度检测层,负责检测图像中的大目标。
      每个检测层会独立输出自身的损失、IOU等统计数据。
  • 每epoch出现四次相关记录:这是因为训练时每个batch会输出两个检测层的结果,你看到的四次是两个batch的检测层数据(日志中total_bbox分别为22和47,对应两个不同的batch)。

二、绘图用数值选择

损失曲线

  • 优先选avg loss:这是整个epoch内所有batch、所有检测层损失的加权平均值,最能体现模型整体的收敛趋势,是绘制训练损失曲线的核心指标。
  • 若需细分不同尺度目标的训练情况:可以分别统计每个epoch内Region 30和Region 37各自的total_loss平均值,绘制两条独立曲线,观察小/大目标的训练收敛差异。
  • 单类别class_loss:仅反映模型对类别的分类精度,若关注分类性能可单独绘制其平均值曲线,但一般训练重点看整体的avg loss。

IOU曲线

日志中每个Region后的Avg (IOU)是对应batch内该检测层预测框与真实框的平均IOU,绘制epoch级IOU曲线需做如下处理:

  • 小目标IOU:计算每个epoch内所有batch中Region 30的Avg (IOU)平均值;
  • 大目标IOU:计算每个epoch内所有batch中Region 37的Avg (IOU)平均值;
  • 整体IOU:按每个batch中两个Region的检测框数量(count字段)加权平均,公式为:(Region30_IOU * Region30_count + Region37_IOU * Region37_count) / (Region30_count + Region37_count),再取整个epoch的加权平均值。

附你提供的日志片段:

在63.765毫秒内加载32张图像
v3 (iou loss, Normalizer: (iou: 0.07, obj: 1.00, cls: 1.00) Region 30 Avg (IOU: 0.500688), count: 12, class_loss = 133.547852, iou_loss = 0.091309, total_loss = 133.639160 
v3 (iou loss, Normalizer: (iou: 0.07, obj: 1.00, cls: 1.00) Region 37 Avg (IOU: 0.490458), count: 10, class_loss = 447.843597, iou_loss = 0.410095, total_loss = 448.253693 
total_bbox=22, rewritten_bbox=0.000000%
v3 (iou loss, Normalizer: (iou: 0.07, obj: 1.00, cls: 1.00) Region 30 Avg (IOU: 0.549514), count: 5, class_loss = 133.405350, iou_loss = 0.028351, total_loss = 133.433701 
v3 (iou loss, Normalizer: (iou: 0.07, obj: 1.00, cls: 1.00) Region 37 Avg (IOU: 0.315040), count: 20, class_loss = 446.940216, iou_loss = 1.281342, total_loss = 448.221558 
total_bbox=47, rewritten_bbox=0.000000%
]2;1/1000: loss=290.4 time=unknown[1;37m1[0m: loss=[0;31m290.437[0m, avg loss=[0;31m290.437[0m, rate=0.000000, 483.468毫秒, 32张图像, time remaining=unknown

内容的提问来源于stack exchange,提问作者1337_N00B

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.29 12:17:05