Caffe CNN特征可视化

转载请注明出处,楼燚(yì)航的blog,http://www.cnblogs.com/louyihang-loves-baiyan/

以下部分代码是根据caffe的python接口,从一次forword中取出param和blob里面的卷积核 和响应的卷积图。

import numpy as np
import matplotlib.pyplot as plt
import os
import caffe
import sys
import pickle
import cv2caffe_root = '../'  deployPrototxt =  '/home/chenjie/louyihang/caffe/models/bvlc_reference_caffenet/deploy_louyihang.prototxt'
modelFile = '/home/chenjie/louyihang/caffe/models/bvlc_reference_caffenet/caffenet_carmodel_louyihang_iter_50000.caffemodel'
meanFile = 'python/caffe/imagenet/ilsvrc_2012_mean.npy'
imageListFile = '/home/chenjie/DataSet/CompCars/data/train_test_split/classification/test_model431_label_start0.txt'
imageBasePath = '/home/chenjie/DataSet/CompCars/data/cropped_image'
resultFile = 'PredictResult.txt'#网络初始化
def initilize():print 'initilize ... 'sys.path.insert(0, caffe_root + 'python')caffe.set_mode_gpu()caffe.set_device(4)net = caffe.Net(deployPrototxt, modelFile,caffe.TEST)return net#取出网络中的params和net.blobs的中的数据
def getNetDetails(image, net):# input preprocessing: 'data' is the name of the input blob == net.inputs[0]transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})transformer.set_transpose('data', (2,0,1))transformer.set_mean('data', np.load(caffe_root + meanFile ).mean(1).mean(1)) # mean pixeltransformer.set_raw_scale('data', 255)  # the reference model operates on images in [0,255] range instead of [0,1]transformer.set_channel_swap('data', (2,1,0))  # the reference model has channels in BGR order instead of RGB# set net to batch size of 50net.blobs['data'].reshape(1,3,227,227)net.blobs['data'].data[...] = transformer.preprocess('data', caffe.io.load_image(image))out = net.forward()#网络提取conv1的卷积核filters = net.params['conv1'][0].datawith open('FirstLayerFilter.pickle','wb') as f:pickle.dump(filters,f)vis_square(filters.transpose(0, 2, 3, 1))#conv1的特征图feat = net.blobs['conv1'].data[0, :36]with open('FirstLayerOutput.pickle','wb') as f:pickle.dump(feat,f)vis_square(feat,padval=1)pool = net.blobs['pool1'].data[0,:36]with open('pool1.pickle','wb') as f:pickle.dump(pool,f)vis_square(pool,padval=1)# 此处将卷积图和进行显示,
def vis_square(data, padsize=1, padval=0 ):data -= data.min()data /= data.max()#让合成图为方n = int(np.ceil(np.sqrt(data.shape[0])))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))#合并卷积图到一个图像中data = data.reshape((n, n) + data.shape[1:]).transpose((0, 2, 1, 3) + tuple(range(4, data.ndim + 1)))data = data.reshape((n * data.shape[1], n * data.shape[3]) + data.shape[4:])print data.shapeplt.imshow(data)if __name__ == "__main__":net = initilize()testimage = '../data/MyTest/visualize_test.jpg'getNetDetails(testimage, net)

输入的测试图像

第一层的卷积核和卷积图,可以看到一些明显的边缘轮廓,左侧是相应的卷积核

第一个Pooling层的特征图

第二层卷积特征图

第二层pooling的特征图,可以看到pooling之后,对conv的特征有部分强化,我网络中使用的max-pooling,但是到了pooling2已经出现一些离散的块了,已经有些抽象了,难以看出什么东西

Caffe CNN特征可视化相关推荐

  1. 4.7 CNN 特征可视化-深度学习第四课《卷积神经网络》-Stanford吴恩达教授

    ←上一篇 ↓↑ 下一篇→ 4.6 什么是神经风格转换? 回到目录 4.8 代价函数 CNN 特征可视化 (What are deep ConvNets learning?) 深度卷积网络到底在学什么? ...

  2. 擦除:提升 CNN 特征可视化的 3 种重要手段

    点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 作者丨皮特潘 导读 所谓擦除,就是去除掉一部分有用的信息,以提高网 ...

  3. 深度 | 谷歌的新CNN特征可视化方法,构造出一个华丽繁复的新世界

    作者:晓凡 概要:近日,来自谷歌大脑和谷歌研究院的一篇技术文章又从一个新的角度拓展了人类对神经网络的理解,得到的可视化结果也非常亮眼.非常魔性. 深度神经网络解释性不好的问题一直是所有研究人员和商业应 ...

  4. 【深度学习】擦除:提升 CNN 特征可视化的 3 种重要手段

    作者丨皮特潘 审稿|邓富城 编辑丨极市平台 导读 所谓擦除,就是去除掉一部分有用的信息,以提高网络提取特征的能力.本文对3种提升特征可视化的方法进行了详细综述,包括直接擦除.利用预测信息(CAM)擦除 ...

  5. pytorch简单代码实现deep dream图(即CNN特征可视化 features visualization)

      本文给出完整代码实现CNN特征的可视化输入图像,也就是简单的deep dream图,有助于更好的理解CNN工作原理,并掌握用梯度上升法生成满足要求输入图像的技术.更清晰美观的deep dream图 ...

  6. caffe预测、特征可视化python接口调用

    转载自: 深度学习(九)caffe预测.特征可视化python接口调用 - hjimce的专栏 - 博客频道 - CSDN.NET http://blog.csdn.net/hjimce/articl ...

  7. 深度学习(九)caffe预测、特征可视化python接口调用

    caffe预测.特征可视化python接口调用 原文地址:http://blog.csdn.net/hjimce/article/details/48972877 作者:hjimce 网上有很多caf ...

  8. CNN训练可视化特征图(tensorflow2.x实现)

    CNN训练可视化(tensorflow2.x实现) 原理介绍 实例化VGG16 加载图片并进行预处理 预测图片 获取指定层对应的输出 可视化CNN训练过程 原理介绍 卷积层由多个卷积核组成,可以将每个 ...

  9. 服不服?40行Python代码,实现卷积特征可视化

    点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 本文转自|深度学习这件小事 卷积神经网络(CNN)变革了计算机视觉 ...

最新文章

  1. 华为SDSec分为几层
  2. Collaborative Filter - Data Mining基础(ACM暑校)
  3. duilib设置滚动条自动滚动到底
  4. Python with和contextlib.closing配合使用(contextlib)
  5. Byval 和 ByRef 使用
  6. 外媒:苹果高管正拜访亚洲供应商,探讨苹果汽车生产事宜
  7. 浅析SQL SERVER执行计划中的各类怪相
  8. python collections模块_python collections模块
  9. 军用装备温湿度循环测试,温湿度环境可靠性实验室GJB150A
  10. 至强服务器性能排行,志强cpu天梯图_2020年至强cpu天梯图性能排行榜
  11. 基于WinPcap的网络流量在线分析系统的设计与实现
  12. PGSQL查询今天生日的员工
  13. 正确安装Senta的姿势
  14. win7任务管理器没有的找到方法
  15. 【AIOT】Python语音处理
  16. 百度旋转验证码(8-24,js逆向)
  17. docker安装青龙面板薅羊毛(新手教程,大佬可略过)
  18. 职中选什么专业好_职高毕业,迷茫中,选什么专业和学校好
  19. 用 LINQ 编写 C# 都有哪些一招必杀的技巧?
  20. 视频批量转GIF工具V1.3免费版

热门文章

  1. 怎么教计算机应用基础,怎样教好《计算机应用基础》课
  2. canvas之一:绘制直线
  3. 计算机开始按钮作用,Win8.1的12个变化:开始按钮回归功能不再
  4. 鸿蒙2.0操作体验,鸿蒙2.0操作系统正式版-华为鸿蒙2.0操作系统正式版官方预约 v1.0.0-优盘手机站...
  5. 补零对有限长序列频谱及DFT的影响
  6. python编程分析了一下高考那些事,发现了两个之最,原来是这样
  7. 外卖ERP管理系统(二)
  8. TypeScript - 不止稳,而且快
  9. ansible-playbook jdk安装
  10. [转]符号和运算符参考 (F#)