shuifuture/yolov7-GradCAM

Implementing GradCAM visualization

Jupyter NotebookGPL-3.0

yolov7-GradCAM

Source: xiao-dragon https://gitee.com/xiao-dragon

介绍

在YOLOv7中实现GradCAM/GradCAM++可视化

具体实现部分可见CSDN博客：【YOLOv7】结合GradCAM热力图可视化

运行

GradCAM

python main_gradcam.py --method gradcam

GradCAM++

python main_gradcam.py --method gradcampp

结果

GradCAM

GradCAM++(待更新...)

参考

https://github.com/pooya-mohammadi/yolov5-gradcam