4
社区成员
发帖
与我相关
我的任务
分享YOLOv5目标检测之Grad-CAM热力图可视化
| 课程名称 | 适应人群 |
|---|---|
| YOLOv5目标检测之Grad-CAM热力图可视化 | 希望学习YOLOv5上的Grad-CAM热力图可视化方法的学员 |
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热力图可视化具体修改的代码讲解。
...全文
请发表友善的回复…
发表回复
车神580 2024-06-13
- 打赏
- 举报
老师您好，我通读了您的代码，同时使用自制的数据集进行实验，发现我输入不同的图片导致激活值和梯度值的向量图尺寸不一样，有的是（16，20）有的是（14，20）不知道原因
杆子提 2023-04-21
- 打赏
- 举报
老师，麻烦问下，运行main文件提升错误：torch.nn.modules.module.ModuleAttributeError: 'SiLU' object has no attribute 'register_full_backward_hook'，该如何解决
qq_54055019 2023-05-09
- 举报
torch的版本不对，我也遇到过这种问题
qq_42542299 2023-08-21
- 举报
请问你解决了吗
m0_52817136 2023-02-23
- 打赏
- 举报
請問如果要使用1280的img size需要更改那一部分？
gdl66 2023-04-14
- 举报
你的热力图可视化实现了吗?
hanhanyld 2022-11-01
- 打赏
- 举报
usage: main.py [-h] [--model-path MODEL_PATH] [--img-path IMG_PATH]
[--output-dir OUTPUT_DIR] [--img-size IMG_SIZE]
[--target-layer TARGET_LAYER] [--method METHOD]
[--device DEVICE] [--names NAMES] [--no_text_box]
main.py: error: unrecognized arguments: model_20_cv3_act
这是什么情况
bai666ai 2022-11-02
- 举报
<p>注意使用课程网盘上的代码</p>
hanhanyld 2022-10-31
- 打赏
- 举报
TypeError: attempt_load() got an unexpected keyword argument 'map_location'
这是什么问题
bai666ai 2022-11-06
- 举报
<p>使用课程网盘上的权重文件试一下</p>
m0_45035146 2022-10-25
- 打赏
- 举报
ValueError: too many values to unpack (expected 2)，当我识别两个变量时候出现了这个错误
m0_45035146 2022-10-25
- 打赏
- 举报
这个训练时候只能单目标吗 而不是多目标识别？
bai666ai 2022-10-26
- 举报
<p>应该可以</p>
m0_45035146 2022-10-22
- 打赏
- 举报
老师，数据集标签应该是个什么样的格式
bai666ai 2022-11-10
- 举报
<p>可以参考课程中提供的数据集</p>
xiahugo 2022-09-26
- 打赏
- 举报
老师，请问课件在哪下下载？
xiahugo 2022-09-26
- 打赏
- 举报
老师，请问课件在哪下下载？
m0_67269661 2022-08-29
- 打赏
- 举报
老师，有没有tensorflow版本的代码？
bai666ai 2022-08-30
- 举报
<p>yolov5是用pytorch开发的，没有tensorflow版本</p>
迪导带的硕士研究生 2022-07-27
- 打赏
- 举报
Traceback (most recent call last):
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\main.py", line 110, in
main(args.img_path)
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\main.py", line 82, in main
saliency_method = YOLOV5GradCAM(model=model, layer_name=args.target_layer, img_size=input_size)
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\models\gradcam.py", line 44, in __init__
self.model(torch.zeros(1, 3, *img_size, device=device))
File "E:\anaconda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\models\yolo_v5_object_detector.py", line 167, in forward
prediction, logits, _ = self.model(img, augment=False)
ValueError: not enough values to unpack (expected 3, got 2)
为啥我运行就报这个错误了呢？还是拿老师的原项目运行的
bai666ai 2022-08-01
- 举报
<p>再检查和对比下项目流程，看是否漏掉了什么</p>
Hbut_xiaobai 2022-07-21
- 打赏
- 举报
老师您好，我用您这个代码可视化自己训练的模型，但是报以下错误，请问是什么原因呢？
Traceback (most recent call last):
File "F:/DeepLearning/Visdrone-finally-experiment/yolov5-6.1-gradcam/main.py", line 110, in
main(args.img_path)
File "F:/DeepLearning/Visdrone-finally-experiment/yolov5-6.1-gradcam/main.py", line 80, in main
torch_img = model.preprocessing(img[..., ::-1])
TypeError: 'NoneType' object is not subscriptable
bai666ai 2022-07-21
- 举报
<p>检查一下你的数据集的标签是否和课程的格式一样</p>
locken1997 2022-06-09
- 打赏
- 举报
老师你好，我点击下载课件后 里面没有代码 请问代码在哪里下载呢
locken1997 2022-06-09
- 打赏
- 举报
老师，请问代码在哪里呢 添加的文件找不到呀
weixin_51971430 2022-06-08
- 打赏
- 举报
老师 请问下为什么我训练的权重文件可视化后冷区显示为目标区域？
bai666ai 2022-06-09
- 举报
<p>注意物体可能由不同的target_layer获得，可修改下试试</p>
<p> </p>
weixin_51971430 2022-06-02
- 打赏
- 举报
老师 请问下添加的项目是在哪下载的 boxes.py等
基于PyTorch的图像分类模型可视化研究：热力图、特征图与Grad-CAM可视化结果对比分析，涉及VGG、ResNet、Densenet及MobileNet网络结构在CIFAR10与Mini-Ima