YOLOv5目标检测之Grad-CAM热力图可视化

深度学习目标检测

Ratings: 4.00 / 5.00

PyTorch版的YOLOv5是一个非常流行的基于深度学习的目标检测器。本课程使用Grad-CAM热力图可视化方法对YOLOv5进行热力图可视化，可以直观展示图像中哪些区域对物体类别分类贡献程度大。

Grad-CAM是一种CNN（卷积神经网络）可解释性的经典方法，与CAM（类激活图）相比，这种方法不需要对模型进行改动就可以生成热力图（heatmap），非常方便和灵活。

本课程在YOLOv5 v6.1版本代码的基础上增加了Grad-CAM热力图可视化方法，演示针对自己的数据集训练和进行Grad-CAM热力图可视化过程，还讲解了原代码针对Grad-CAM热力图可视化的所做的修改部分。

本课程分为原理篇、项目实战篇、代码讲解篇。

· 原理篇包括：Grad-CAM热力图可视化原理。

· 项目实战篇包括：PyTorch环境安装、YOLOv5项目安装、准备自己的数据集、修改配置文件、训练自己的数据集、Grad-CAM热力图可视化。

· 代码讲解篇包括：针对Grad-CAM热力图可视化具体修改的代码讲解。