一、本文介绍
本文带来的是RT-DETR热力图可视化功能，支持指定模型，指定显示层，设置置信度，以及10种可视化实现方式。
我们经常看到一些论文里绘制了不同的热力图，一方面能够直观的感受其模型的有效性，另一方面也丰富了论文内容。特别是在使用了注意力模块的网络中，热力图就可以验证注意力机制是否真正聚焦到了预期的重要特征上，以便对模型的有效性和合理性进行评估。例如Centralized Feature Pyramid for Object Detection这篇文章中展示的，就很能够表达作者改进后的模型相比之前模型的优越性。
文章目录
二、项目完整代码
在项目根目录中新建heatmap.py文件，并粘贴如下代码：
项目中主要使用grad_cam这个依赖