在这里插入图片描述

这张图片是一个深度学习模型训练完成后,项目文件夹里常见的一些可视化输出文件(通常是 TensorBoard 或自定义 logger 生成的),它们的作用是帮助我们快速诊断模型训练得怎么样、是否存在问题。下面我按类别逐一详细解释每张图的含义和作用:

  1. weights(文件夹图标)
    保存模型的权重文件(.pth 或 .ckpt),用于后续加载模型推理或继续训练。

  2. args.yaml
    记录本次训练的所有超参数(学习率、batch_size、优化器、数据路径等),非常重要,方便复现实验。

  3. BoxF1_curve、BoxP_curve、BoxR_curve、BoxR_curve(重复)
    这四张是目标检测任务(通常是 YOLO 系列)的核心指标曲线:

    • BoxF1_curve:F1 分数随 confidence 阈值变化的曲线,综合精度和召回率
    • BoxP_curve:Precision(精确率)曲线
    • BoxR_curve:Recall(召回率)曲线
    • BoxPR_curve:PR 曲线(Precision-Recall Curve),曲线下面积越大越好
      作用:判断模型在不同置信度阈值下的表现,是否过拟合、是否存在精度-召回率严重失衡。
  4. confusion_matrix 和 confusion_matrix_normalized
    混淆矩阵(通常是分类任务用的,但这里可能是类别级别的检测混淆):

    • 左图:未归一化的混淆矩阵(绝对数量)
    • 右图:归一化后的混淆矩阵(每一行和为1)
      作用:直观看出模型把哪些类别搞混了(比如把“猫”错分成“狗”很多),是诊断类别不平衡和特征相似问题的利器。
  5. labels(可能是 labels.jpg 或 labels_correlation.jpg)
    通常是各类别之间标签相关性的热力图,或者训练集中各类别样本数量分布图。
    作用:检查数据是否存在严重类别不平衡,或者某些类别天然高度相关(会导致误分类)。

  6. results(文件夹或 results.png)
    通常包含最终的 mAP、mAP@50、mAP@75 等汇总指标表格,或者 best.pt / last.pt 模型文件。

  7. train_batch0/1/2.jpg
    训练集某个 batch 的可视化结果:

    • 左下角小图:原始图像 + 真实标注框(ground truth)
    • 大图:模型预测的 bounding box(带置信度和类别)
      作用:肉眼检查模型在训练集上是否已经学得差不多,框得准不准、类别标得对不对,漏检多不多。
  8. val_batch0_labels / val_batch0_pred / val_batch1_labels 等
    验证集多个 batch 的预测结果对比:

    • 带 “_labels” 后缀:真实标注(ground truth)
    • 带 “_pred” 后缀:模型预测结果
      作用:最直观的定性分析!可以一眼看出:
    • 模型在验证集上漏检严重吗?
    • 小目标检测能力如何?
    • 是否存在大量误检(蓝框乱标)?
    • 边界框定位准不准(IoU 高不高)?
    • 某些场景(如田地边缘、光照变化)是否容易失效?
  9. val_batchX_labe ls(拼写错误,应该是 labels)
    同上,真实标注图。

总结这些图的整体诊断作用(看完这堆图基本能判断训练好坏):

看什么问题 优先看哪几张图
指标是否达到预期 BoxPR_curve、F1_curve、results 表格
是否过拟合 训练集 batch 图(基本完美) vs 验证集 batch 图(很差)
漏检严重 验证集 pred 图里很多真实框没有对应的预测框
误检/背景错当目标 验证集 pred 图里出现大量没有真实框的蓝框
小目标检测差 验证集图里小物体几乎没框出来
类别混淆 confusion_matrix(对角线外高亮区域)
定位不准(框歪、太大太小) 验证集 pred 和 labels 对齐对比
数据不平衡 labels 相关性图或各类别 AP 差异巨大

一句话总结:
这套图就是 YOLO 系列(尤其是 Ultralytics YOLOv5/v8)训练完成后自动生成的“训练体检报告”。只要把验证集的几张 pred/labels 图和 PR 曲线、混淆矩阵一起看,3 分钟就能判断这次训练是成功、基本可用、还是彻底寄了,需要换骨干网络、加数据、调超参或重标注。非常实用!

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐