YOLOv10 双分支模型HeatMap热力图开发
一、开发背景
之前在使用YOLOv10构建目标检测模型的时候一直希望可以有一个热力图可视化分析的功能,但是官方项目好像一直没有支持完善这块的内容,自己也没有去折腾后面就搁置了,最近又用到了YOLOv10模型来做实验,就想着趁此机会把这块给搞一下。本次开发的目标,是在官方yolov10-main项目上增加一套热力图分析可视化能力,使 YOLOv10 不仅可以输出检测框,还可以展示模型在中间特征层中更关注图像的哪些区域。
最终实现的核心能力是:
- 保留官方 YOLOv10 原有训练、验证、推理流程。
- 新增独立的 EigenCAM 热力图模块。
- 支持 YOLOv10 双分支检测头结构。
- 支持通过命令行或 Python API 生成热力图。
- 输出三类结果:热力图叠加图、灰度热力图、检测框可视化图。
新增文件如下:
yolov10-main/ ├── yolov10_cam_demo.py ├── docs/ │ └── cn/ │ └── yolov10_heatmap.md └── ultralytics/ ├── yolo_cam/ │ ├── __init__.py │ └── eigen_cam.py └── models/ └── yolov10/ ├── __init__.py └── heatmap.py