Keras中可视化卷积层的类激活热力图

【时间】2018.12.26

【题目】Keras中可视化卷积层的类激活热力图

概述    

    本文是对《Deep Learning with python》一书中第5.4.3节Keras中可视化类激活的热力图的整理与总结,参考链接:https://blog.youkuaiyun.com/einstellung/article/details/82858974

一、什么是类激活图(CAP)?

    类激活图(CAM,class activation map)是与特定输出类别相关的二维分数网格,对于输入图像的每个位置进行计算,它表示每个位置对该类别的重要程度。

二、类激活图可视化

    类激活图可视化有助于了解一张图像的哪一部分让卷积神经网络做出了最终的分类决策。这有助于对卷积神经网络的决策过程进行调试,特别是分类错误的情况下。同时,这种方法可以定位图像中的特定目标。这里,我们介绍的具体实现方式是“Grad-CAM:visual explanations from deep networks via gradiaent-based localization',它是在2017年提出来的,具体实现方法是:给定一张输入图像,对于一个卷积层的输出特征图,用类别相对于每一个通道的梯度对这个特征图中的每个通道进行加权。

三、使用VGG16演示类激活图可视化

3.1 加载预训练的VGG-16:

from keras.applications.vgg16 import VGG16

model = VGG16(weights='imagenet')

model.summary()

【VGG-16】模型结构如下:最后输出的是1000维的向量,表示属于1000个类别的概率

3.2 载入输入图像及预处理

如图所示,这是两只非洲象的图片。我们将这张图片转换为VGG16能够读取的格式:大小为224X224的图像,这些训练图

### 像分热力可视化方法 对于像分模型生成的热力进行可视化,主要目的是理解模型对输入像各个部分的关注程度。这有助于解释模型决策过程并提高其透明度。 #### 使用Grad-CAM算法 一种常用的方法是采用梯度加权激活映射(Grad-CAM)[^2]。该技术利用目标别相对于卷积特征的梯度来定位重要区域。具体来说,在前向传播之后计算损失函数关于最后一个卷积层输出的导数;接着取平均值得到权重系数,并将其乘回对应的特征响应上得到最终的热力表示。 ```python from tensorflow.keras.models import Model import numpy as np import cv2 def get_gradcam(model, img_array, layer_name='block5_conv3'): grad_model = Model([model.inputs], [model.get_layer(layer_name).output, model.output]) with tf.GradientTape() as tape: conv_outputs, predictions = grad_model(img_array) loss = predictions[:, np.argmax(predictions[0])] grads = tape.gradient(loss, conv_outputs) pooled_grads = K.mean(grads, axis=(0, 1, 2)) heatmap = tf.reduce_mean(tf.multiply(pooled_grads, conv_outputs), axis=-1) heatmap = np.maximum(heatmap, 0) max_heat = np.max(heatmap) if max_heat == 0: max_heat = 1e-10 heatmap /= max_heat return heatmap.numpy()[0] # 加载预训练模型和测试片 image_path = 'test_image.jpg' img = preprocess_input(load_img(image_path)) # 获取热力 heat_map = get_gradcam(pretrained_model, img) # 将热力叠加到原始片上 original_img = cv2.imread(image_path) overlayed_img = overlay_heatmap(original_img, heat_map) cv2.imshow('Heat Map', overlayed_img) ``` 上述代码展示了如何使用`tensorflow`框架中的工具实现简单的Grad-CAM功能[^2]。需要注意的是,这里假设已经有一个预先训练好的像分器(`pretrained_model`)以及一张待处理的测试片(`image_path`)。 为了更好地呈现效果,还可以考虑调整参数或者尝试其他变体如Guided Grad-CAM等更高级的技术。
评论 21
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值