【深度学习系列】CNN模型的可视化

时间:2021-08-13 02:40:52

  前面几篇文章讲到了卷积神经网络CNN,但是对于它在每一层提取到的特征以及训练的过程可能还是不太明白,所以这节主要通过模型的可视化来神经网络在每一层中是如何训练的。我们知道,神经网络本身包含了一系列特征提取器,理想的feature map应该是稀疏的以及包含典型的局部信息。通过模型可视化能有一些直观的认识并帮助我们调试模型,比如:feature map与原图很接近,说明它没有学到什么特征;或者它几乎是一个纯色的图,说明它太过稀疏,可能是我们feature map数太多了(feature_map数太多也反映了卷积核太小)。可视化有很多种,比如:feature map可视化、权重可视化等等,我以feature map可视化为例。

 


 模型可视化

  因为我没有搜到用paddlepaddle在imagenet 1000分类的数据集上预训练好的googLeNet inception v3,所以用了keras做实验,以下图作为输入:

  • 输入图片 
    • 北汽绅宝D50:

【深度学习系列】CNN模型的可视化

  • feature map可视化 

  取网络的前15层,每层取前3个feature map。

  北汽绅宝D50 feature map:

  【深度学习系列】CNN模型的可视化

  从左往右看,可以看到整个特征提取的过程,有的分离背景、有的提取轮廓,有的提取色差,但也能发现10、11层中间两个feature map是纯色的,可能这一层feature map数有点多了,另外北汽绅宝D50的光晕对feature map中光晕的影响也能比较明显看到。

  • Hypercolumns 
    通常我们把神经网络最后一个fc全连接层作为整个图片的特征表示,但是这一表示可能过于粗糙(从上面的feature map可视化也能看出来),没法精确描述局部空间上的特征,而网络的第一层空间特征又太过精确,缺乏语义信息(比如后面的色差、轮廓等),于是论文《Hypercolumns for Object Segmentation and Fine-grained Localization》提出一种新的特征表示方法:Hypercolumns——将一个像素的 hypercolumn 定义为所有 cnn 单元对应该像素位置的激活输出值组成的向量),比较好的tradeoff了前面两个问题,直观地看如图:

【深度学习系列】CNN模型的可视化

 

  把北汽绅宝D50 第1、4、7层的feature map以及第1, 4, 7, 10, 11, 14, 17层的feature map分别做平均,可视化如下:

  【深度学习系列】CNN模型的可视化


 

代码实践

  1 # -*- coding: utf-8 -*-
2 from keras.applications import InceptionV3
3 from keras.applications.inception_v3 import preprocess_input
4 from keras.preprocessing import image
5 from keras.models import Model
6 from keras.applications.imagenet_utils import decode_predictions
7 import numpy as np
8 import cv2
9 from cv2 import *
10 import matplotlib.pyplot as plt
11 import scipy as sp
12 from scipy.misc import toimage
13
14 def test_opencv():
15 # 加载摄像头
16 cam = VideoCapture(0) # 0 -> 摄像头序号,如果有两个三个四个摄像头,要调用哪一个数字往上加嘛
17 # 抓拍 5 张小图片
18 for x in range(0, 5):
19 s, img = cam.read()
20 if s:
21 imwrite("o-" + str(x) + ".jpg", img)
22
23 def load_original(img_path):
24 # 把原始图片压缩为 299*299大小
25 im_original = cv2.resize(cv2.imread(img_path), (299, 299))
26 im_converted = cv2.cvtColor(im_original, cv2.COLOR_BGR2RGB)
27 plt.figure(0)
28 plt.subplot(211)
29 plt.imshow(im_converted)
30 return im_original
31
32 def load_fine_tune_googlenet_v3(img):
33 # 加载fine-tuning googlenet v3模型,并做预测
34 model = InceptionV3(include_top=True, weights='imagenet')
35 model.summary()
36 x = image.img_to_array(img)
37 x = np.expand_dims(x, axis=0)
38 x = preprocess_input(x)
39 preds = model.predict(x)
40 print('Predicted:', decode_predictions(preds))
41 plt.subplot(212)
42 plt.plot(preds.ravel())
43 plt.show()
44 return model, x
45
46 def extract_features(ins, layer_id, filters, layer_num):
47 '''
48 提取指定模型指定层指定数目的feature map并输出到一幅图上.
49 :param ins: 模型实例
50 :param layer_id: 提取指定层特征
51 :param filters: 每层提取的feature map数
52 :param layer_num: 一共提取多少层feature map
53 :return: None
54 '''
55 if len(ins) != 2:
56 print('parameter error:(model, instance)')
57 return None
58 model = ins[0]
59 x = ins[1]
60 if type(layer_id) == type(1):
61 model_extractfeatures = Model(input=model.input, output=model.get_layer(index=layer_id).output)
62 else:
63 model_extractfeatures = Model(input=model.input, output=model.get_layer(name=layer_id).output)
64 fc2_features = model_extractfeatures.predict(x)
65 if filters > len(fc2_features[0][0][0]):
66 print('layer number error.', len(fc2_features[0][0][0]),',',filters)
67 return None
68 for i in range(filters):
69 plt.subplots_adjust(left=0, right=1, bottom=0, top=1)
70 plt.subplot(filters, layer_num, layer_id + 1 + i * layer_num)
71 plt.axis("off")
72 if i < len(fc2_features[0][0][0]):
73 plt.imshow(fc2_features[0, :, :, i])
74
75 # 层数、模型、卷积核数
76 def extract_features_batch(layer_num, model, filters):
77 '''
78 批量提取特征
79 :param layer_num: 层数
80 :param model: 模型
81 :param filters: feature map数
82 :return: None
83 '''
84 plt.figure(figsize=(filters, layer_num))
85 plt.subplot(filters, layer_num, 1)
86 for i in range(layer_num):
87 extract_features(model, i, filters, layer_num)
88 plt.savefig('sample.jpg')
89 plt.show()
90
91 def extract_features_with_layers(layers_extract):
92 '''
93 提取hypercolumn并可视化.
94 :param layers_extract: 指定层列表
95 :return: None
96 '''
97 hc = extract_hypercolumn(x[0], layers_extract, x[1])
98 ave = np.average(hc.transpose(1, 2, 0), axis=2)
99 plt.imshow(ave)
100 plt.show()
101
102 def extract_hypercolumn(model, layer_indexes, instance):
103 '''
104 提取指定模型指定层的hypercolumn向量
105 :param model: 模型
106 :param layer_indexes: 层id
107 :param instance: 模型
108 :return:
109 '''
110 feature_maps = []
111 for i in layer_indexes:
112 feature_maps.append(Model(input=model.input, output=model.get_layer(index=i).output).predict(instance))
113 hypercolumns = []
114 for convmap in feature_maps:
115 for i in convmap[0][0][0]:
116 upscaled = sp.misc.imresize(convmap[0, :, :, i], size=(299, 299), mode="F", interp='bilinear')
117 hypercolumns.append(upscaled)
118 return np.asarray(hypercolumns)
119
120 if __name__ == '__main__':
121 img_path = '~/auto1.jpg'
122 img = load_original(img_path)
123 x = load_fine_tune_googlenet_v3(img)
124 extract_features_batch(15, x, 3)
125 extract_features_with_layers([1, 4, 7])
126 extract_features_with_layers([1, 4, 7, 10, 11, 14, 17])

 


 

总结

  还有一些网站做的关于CNN的可视化做的非常不错,譬如这个网站:http://shixialiu.com/publications/cnnvis/demo/,大家可以在训练的时候采取不同的卷积核尺寸和个数对照来看训练的中间过程。最近PaddlePaddle也开源了可视化工具VisaulDL,下篇文章我们讲讲paddlepaddle的visualDL和tesorflow的tensorboard。