# yolov5_GradCAM

**Repository Path**: kay545/yolov5_GradCAM

## Basic Information

- **Project Name**: yolov5_GradCAM
- **Description**: 在YOLOv5-6.1版本中，实现GradCAM可视化
- **Primary Language**: Python
- **License**: GPL-3.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 2
- **Created**: 2024-06-16
- **Last Updated**: 2024-06-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# yolov5-6.1-GradCAM

#### 介绍
在YOLOv5-6.1版本中，中实现GradCAM/GradCAM++可视化

具体实现部分可见CSDN博客：[【YOLOv5】结合GradCAM热力图可视化](https://blog.csdn.net/weixin_43799388/article/details/126207632)

#### 运行
GradCAM
```bash
python main_gradcam.py --method gradcam
```

GradCAM++
```bash
python main_gradcam.py --method gradcampp
```

#### 结果
GradCAM

<img src="figure/eagle_result.jpg" width="100%" >

&nbsp;
<img src="figure/cat-dog_result.jpg" width="100%" >

&nbsp;
<img src="figure/dog_result.jpg" width="100%" >



&nbsp;
GradCAM++(待更新...)


#### 参考
https://github.com/pooya-mohammadi/yolov5-gradcam