gradcam: Class Activation Maps for Semantic Segmentation

CVer儿

于 2024-04-10 17:05:00 发布

阅读量96

点赞数

分类专栏：语义分割文章标签： python

本文链接：https://blog.csdn.net/qq_35054151/article/details/137603089

版权

语义分割专栏收录该内容

121 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了GradCAM技术，一种用于语义分割的可视化工具，通过类激活映射揭示图像中影响分类的区域。GradCAM利用梯度信息来定位目标类别在特征图上的重要区域，为深度学习模型的决策过程提供了直观的理解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pip install grad-cam -i https://pypi.tuna.tsinghua.edu.cn/simple

import warnings
warnings.filterwarnings('ignore')
warnings.simplefilter('ignore')
from torchvision.models.segmentation import deeplabv3_resnet50
import torch
import torch.functional as F
import numpy as np
import requests
import torchvision
from PIL import Image
from pytorch_grad_cam.utils.image import show_cam_on_image, preprocess_image


image_url = "https://farm1.staticflickr.com/6/9606553_ccc7518589_z.jpg"
image = np.array(Image.open(requests.get(image_url, stream=True).raw))
rgb_img = np.floa