Caffe 的可视化（四）特征图的可视化

最新推荐文章于 2023-08-04 22:35:57 发布

sushiqian

最新推荐文章于 2023-08-04 22:35:57 发布

阅读量2k

点赞数

CC 4.0 BY-SA版权

分类专栏： Caffe 文章标签： caffe

本文链接：https://blog.youkuaiyun.com/sushiqian/article/details/78614133

Caffe 的可视化（四）特征图的可视化

接着上一篇（http://blog.youkuaiyun.com/sushiqian/article/details/78613726）

输入model 一张图片，查看它经过卷积后生成的特征图

编写代码 extract_data.py 如下：


import numpy as np
import matplotlib.pyplot as plt
import os
import sys
import caffe

CAFFE_ROOT = '/home/mark/caffe'
deploy_file_name = 'cifar10_deploy.prototxt'
model_file_name  = 'cifar10_quick_iter_4000.caffemodel'
test_img   = "./cat.jpg"

#编写一个函数，用于显示各层的参数,padsize用于设置图片间隔空隙,padval用于调整亮度 
def show_data(data, padsize=1, padval=0, name="feature-data.jpg"):
    #归一化
    data -= data.min()
    data /= data.max()

    #根据data中图片数量data.shape[0]，计算最后输出时每行每列图片数n
    n = int(np.ceil(np.sqrt(data.shape[0])))
    # padding = ((图片个数维度的padding),(图片高的padding), (图片宽的padding), ....)
    padding = ((0, n ** 2 - data.shape[0]), (0, padsize), (0, padsize)) + ((0, 0),) * (data.ndim - 3)
    data = np.pad(data, padding, mode='constant', constant_values=(padval, padval))

    # 先将padding后的da