当前位置:   article > 正文

第55步 深度学习图像识别:CNN特征层和卷积核可视化(TensorFlow)_第55步 深度学习图像识别:cnn特征层和卷积核可视化(tensorflow)

第55步 深度学习图像识别:cnn特征层和卷积核可视化(tensorflow)

基于WIN10的64位系统演示

一、写在前面

(1)CNN可视化

在理解和解释卷积神经网络(CNN)的行为方面,可视化工具起着重要的作用。以下是一些可以用于可视化的内容:

(a)激活映射(Activation maps):可以显示模型在训练过程中的激活情况,这可以帮助我们理解每一层(或每个过滤器)在识别图像的哪个部分。

(b)过滤器(Filter):可以将卷积核或过滤器的权重可视化,以了解模型在寻找什么样的特征。

(c)类激活映射(Class Activation Mapping,CAM):这是一种可视化技术,可以用于理解网络为什么会将图像分类到特定类别中。这通过生成一个热图来实现,该热图高亮显示网络认为对分类决策最重要的区域。

(d)混淆矩阵(Confusion Matrix):在分类问题中,混淆矩阵可以帮助我们理解模型在各个类别上的性能,显示模型何时正确分类,何时出现错误。

(e)学习曲线(Learning Curves):显示训练和验证误差随着训练周期(或时间)的变化,可以帮助我们理解模型是否过拟合,是否需要更多的训练等。

(2)激活映射和过滤器

由于混淆矩阵和学习曲线之前都已经给了代码,这一期主要介绍激活映射和过滤器(卷积层)。用的模型是Mobilenet_v2,以为够快!!

二、CNN可视化实战

继续使用胸片的数据集:肺结核病人和健康人的胸片的识别。其中,肺结核病人700张,健康人900张,分别存入单独的文件夹中。

(a)Mobilenet_v2建模

  1. ######################################导入包###################################
  2. from tensorflow import keras
  3. import tensorflow as tf
  4. from tensorflow.python.keras.layers import Dense, Flatten, Conv2D, MaxPool2D, Dropout, Activation, Reshape, Softmax, GlobalAveragePooling2D, BatchNormalization
  5. from tensorflow.python.keras.layers.convolutional import Convolution2D, MaxPooling2D
  6. from tensorflow.python.keras import Sequential
  7. from tensorflow.python.keras import Model
  8. from tensorflow.python.keras.optimizers import adam_v2
  9. import numpy as np
  10. import matplotlib.pyplot as plt
  11. from tensorflow.python.keras.preprocessing.image import ImageDataGenerator, image_dataset_from_directory
  12. from tensorflow.python.keras.layers.preprocessing.image_preprocessing import RandomFlip, RandomRotation, RandomContrast, RandomZoom, RandomTranslation
  13. import os,PIL,pathlib
  14. import warnings
  15. #设置GPU
  16. gpus = tf.config.list_physical_devices("GPU")
  17. if gpus:
  18. gpu0 = gpus[0] #如果有多个GPU,仅使用第0个GPU
  19. tf.config.experimental.set_memory_growth(gpu0, True) #设置GPU显存用量按需使用
  20. tf.config.set_visible_devices([gpu0],"GPU")
  21. warnings.filterwarnings("ignore") #忽略警告信息
  22. plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
  23. plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
  24. ################################导入数据集#####################################
  25. #1.导入数据
  26. data_dir = "./MTB"
  27. data_dir = pathlib.Path(data_dir)
  28. image_count = len(list(data_dir.glob('*/*')))
  29. print("图片总数为:",image_count)
  30. batch_size = 32
  31. img_height = 100
  32. img_width = 100
  33. train_ds = image_dataset_from_directory(
  34. data_dir,
  35. validation_split=0.2,
  36. subset="training",
  37. seed=12,
  38. image_size=(img_height, img_width),
  39. batch_size=batch_size)
  40. val_ds = image_dataset_from_directory(
  41. data_dir,
  42. validation_split=0.2,
  43. subset="validation",
  44. seed=12,
  45. image_size=(img_height, img_width),
  46. batch_size=batch_size)
  47. class_names = train_ds.class_names
  48. print(class_names)
  49. print(train_ds)
  50. #2.检查数据
  51. for image_batch, labels_batch in train_ds:
  52. print(image_batch.shape)
  53. print(labels_batch.shape)
  54. break
  55. #3.配置数据
  56. AUTOTUNE = tf.data.AUTOTUNE
  57. def train_preprocessing(image,label):
  58. return (image/255.0,label)
  59. train_ds = (
  60. train_ds.cache()
  61. .shuffle(800)
  62. .map(train_preprocessing)
  63. .prefetch(buffer_size=AUTOTUNE)
  64. )
  65. val_ds = (
  66. val_ds.cache()
  67. .map(train_preprocessing)
  68. .prefetch(buffer_size=AUTOTUNE)
  69. )
  70. #4. 数据可视化
  71. plt.figure(figsize=(10, 8)) # 图形的宽为10高为5
  72. plt.suptitle("数据展示")
  73. class_names = ["Tuberculosis","Normal"]
  74. for images, labels in train_ds.take(1):
  75. for i in range(15):
  76. plt.subplot(4, 5, i + 1)
  77. plt.xticks([])
  78. plt.yticks([])
  79. plt.grid(False)
  80. # 显示图片
  81. plt.imshow(images[i])
  82. # 显示标签
  83. plt.xlabel(class_names[labels[i]-1])
  84. plt.show()
  85. ######################################数据增强函数################################
  86. data_augmentation = Sequential([
  87. RandomFlip("horizontal_and_vertical"),
  88. RandomRotation(0.2),
  89. RandomContrast(1.0),
  90. RandomZoom(0.5,0.2),
  91. RandomTranslation(0.3,0.5),
  92. ])
  93. def prepare(ds):
  94. ds = ds.map(lambda x, y: (data_augmentation(x, training=True), y), num_parallel_calls=AUTOTUNE)
  95. return ds
  96. train_ds = prepare(train_ds)
  97. ################################导入mobilenet_v2################################
  98. #获取预训练模型对输入的预处理方法
  99. from tensorflow.python.keras.applications import mobilenet_v2
  100. from tensorflow.python.keras import Input, regularizers
  101. IMG_SIZE = (img_height, img_width, 3)
  102. # 创建输入张量
  103. inputs = Input(shape=IMG_SIZE)
  104. # 定义基础模型,并将 inputs 传入
  105. base_model = mobilenet_v2.MobileNetV2(input_tensor=inputs,
  106. include_top=False,
  107. weights='imagenet')
  108. #从基础模型中获取输出
  109. x = base_model.output
  110. #全局池化
  111. x = GlobalAveragePooling2D()(x)
  112. #BatchNormalization
  113. x = BatchNormalization()(x)
  114. #Dropout
  115. x = Dropout(0.8)(x)
  116. #Dense
  117. x = Dense(128, kernel_regularizer=regularizers.l2(0.1))(x) # 全连接层减少到128,添加 L2 正则化
  118. #BatchNormalization
  119. x = BatchNormalization()(x)
  120. #激活函数
  121. x = Activation('relu')(x)
  122. #输出层
  123. outputs = Dense(2, kernel_regularizer=regularizers.l2(0.1))(x) # 添加 L2 正则化
  124. #BatchNormalization
  125. outputs = BatchNormalization()(outputs)
  126. #激活函数
  127. outputs = Activation('sigmoid')(outputs)
  128. #整体封装
  129. model = Model(inputs, outputs)
  130. #打印模型结构
  131. print(model.summary())
  132. #############################编译模型#########################################
  133. #定义优化器
  134. from tensorflow.python.keras.optimizers import adam_v2, rmsprop_v2
  135. optimizer = adam_v2.Adam()
  136. #编译模型
  137. model.compile(optimizer=optimizer,
  138. loss='sparse_categorical_crossentropy',
  139. metrics=['accuracy'])
  140. #训练模型
  141. from tensorflow.python.keras.callbacks import ModelCheckpoint, Callback, EarlyStopping, ReduceLROnPlateau, LearningRateScheduler
  142. NO_EPOCHS = 50
  143. PATIENCE = 10
  144. VERBOSE = 1
  145. # 设置动态学习率
  146. annealer = LearningRateScheduler(lambda x: 1e-5 * 0.99 ** (x+NO_EPOCHS))
  147. # 设置早停
  148. earlystopper = EarlyStopping(monitor='loss', patience=PATIENCE, verbose=VERBOSE)
  149. #
  150. checkpointer = ModelCheckpoint('mtb_jet_best_model_mobilenetv3samll.h5',
  151. monitor='val_accuracy',
  152. verbose=VERBOSE,
  153. save_best_only=True,
  154. save_weights_only=True)
  155. train_model = model.fit(train_ds,
  156. epochs=NO_EPOCHS,
  157. verbose=1,
  158. validation_data=val_ds,
  159. callbacks=[earlystopper, checkpointer, annealer])

(b)将前M层每个中间激活的所有通道可视化

  1. from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
  2. from tensorflow.python.keras import Model
  3. import matplotlib.pyplot as plt
  4. from tensorflow.python.keras.models import load_model
  5. from tensorflow.python.keras.preprocessing import image
  6. from tensorflow.python.keras.preprocessing.image import img_to_array
  7. from PIL import Image
  8. if __name__=='__main__':
  9. #加载保存的模型
  10. model=load_model('mtb_jet_best_model_mobilenet.h5')
  11. model.summary()
  12. #加载一张猫的图像
  13. img=image.load_img(path='./MTB/Tuberculosis/Tuberculosis-203.png',target_size=(img_width,img_height))
  14. img_tensor=img_to_array(img)
  15. img_tensor=img_tensor.reshape((1,)+img_tensor.shape)
  16. img_tensor/=255.
  17. plt.imshow(img_tensor[0])
  18. plt.show()
  19. N=25#提取前N层的输出
  20. layer_outputs=[layer.output for layer in model.layers[:N]]
  21. activation_model=Model(inputs=model.input,outputs=layer_outputs)
  22. #以预测模式运行模型 activations包含卷积层的N个输出
  23. activations=activation_model.predict(img_tensor)
  24. print(activations[0].shape)
  25. first_layer_activation = activations[0]
  26. plt.matshow(first_layer_activation[0, :, :, 1], cmap='viridis')
  27. plt.show()
  28. #清空当前图像
  29. #plt.clf()
  30. #将每个中间激活的所有通道可视化
  31. layer_names = []
  32. for layer in model.layers[:N]:
  33. layer_names.append(layer.name)
  34. images_per_row = 16
  35. for layer_name, layer_activation in zip(layer_names, activations):
  36. n_features = layer_activation.shape[-1]
  37. size = layer_activation.shape[1]
  38. n_cols = n_features // images_per_row
  39. display_grid = np.zeros((size * n_cols, images_per_row * size))
  40. for col in range(n_cols):
  41. for row in range(images_per_row):
  42. channel_image = layer_activation[0,:, :,col * images_per_row + row]
  43. channel_image -= channel_image.mean()
  44. channel_image /= channel_image.std()
  45. channel_image *= 64
  46. channel_image += 128
  47. channel_image = np.clip(channel_image, 0, 255).astype('uint8')
  48. display_grid[col * size : (col + 1) * size,
  49. row * size : (row + 1) * size] = channel_image
  50. scale = 1. / size
  51. plt.figure(figsize=(scale * display_grid.shape[1],
  52. scale * display_grid.shape[0]))
  53. plt.title(layer_name)
  54. plt.grid(False)
  55. plt.imshow(display_grid, aspect='auto', cmap='viridis')
  56. plt.show()

这个代码输出了模型的前25层的所有中间的激活图层。

首先我们来看看模型结构,非常长的一个表格:

 注意哈,第一层是输入层,所以后面是不会输出的。只会输出后24个图层:

先来看前六个图层的处理:

(1)Conv1 (Conv2D): 这是一个卷积层,它对输入图像进行卷积操作以提取图像的局部特征。在这个例子中,这个层级的输出形状为(50, 50, 32),表示该层使用了32个卷积核,生成了32个特征图(feature map),每个特征图的尺寸是50x50。

(2)bn_Conv1 (BatchNormalization): 这是一个批标准化层,它将卷积层的输出进行标准化处理,使得结果的均值接近0,标准差接近1。这个操作可以提高模型的训练速度和模型性能。

(3)Conv1_relu (ReLU): 这是一个ReLU激活函数,它对上一层的输出应用非线性变换(将所有负值变为0)。这样可以增加模型对复杂模式的学习能力。

(4)expanded_conv_depthwise (DepthwiseConv2D): 这是一个深度可分离卷积层,它首先对每个输入通道分别进行卷积操作,然后再通过1x1的卷积将这些结果进行组合。这种操作相比传统的卷积操作可以减少模型的参数和计算量。

(5-6)expanded_conv_depthwise_BN (BatchNormalization) 和 expanded_conv_depthwise_relu (ReLU): 这两个层和上面的bn_Conv1、Conv1_relu层的操作是类似的,也是分别对深度卷积层的结果进行标准化和ReLU非线性变换。

接着,往后看:

继续解读:

(7)expanded_conv_project (Conv2D): 这是一个卷积层,它对上一层的深度卷积层的输出进行处理,输出形状为(50, 50, 16)。

(8)expanded_conv_project_BN (BatchNormalization): 这是一个批标准化层,对卷积层的输出进行标准化处理。

(9-10)block_1_expand (Conv2D) 和 block_1_expand_BN (BatchNormalization): 这是另一个卷积层和批标准化层,它们对上一层的输出进行卷积操作和标准化操作,输出形状为(50, 50, 96)。

(11)block_1_expand_relu (ReLU): 这是一个ReLU激活函数,它对上一层的输出应用非线性变换(将所有负值变为0)。

(12)block_1_pad (ZeroPadding2D): 这是一个零填充层,它在输入的周围填充零以保持特征图的空间尺寸。

(13-14)block_1_depthwise (DepthwiseConv2D) 和 block_1_depthwise_BN (BatchNormalization): 这是深度可分离卷积层和批标准化层,它们对上一层的输出进行深度卷积操作和标准化操作,输出形状为(25, 25, 96)。

(15)block_1_depthwise_relu (ReLU): 这又是一个ReLU激活函数,它对上一层的输出应用非线性变换。

 (16-17)block_1_project (Conv2D) 和 block_1_project_BN (BatchNormalization): 这是卷积层和批标准化层,它们对上一层的输出进行卷积操作和标准化操作,输出形状为(25, 25, 24)。

(18-24)接下来的block_2_expand (Conv2D) 到 block_2_project (Conv2D) 层是另一个与block_1相似的模块,只不过输入形状改变了。它们进行的操作和上述的操作大致相同,也是包含卷积、批标准化和ReLU激活等操作。

注意看:以上每个层级都在处理和转化上一层的输出,以便提取和学习图像的特征。这样的架构可以帮助模型捕捉到更复杂的模式和特征。通俗来说,就是随着层数增加,就越能看到细节!其中黄色的部分是模型重点关注的区域!

(b)显示第M层输出的第L个通道图像

  1. ##加载保存的模型
  2. from tensorflow.python.keras.models import load_model
  3. from tensorflow.python.keras.preprocessing import image
  4. from tensorflow.python.keras.preprocessing.image import img_to_array
  5. from PIL import Image
  6. if __name__=='__main__':
  7. model=load_model('mtb_jet_best_model_mobilenet.h5')
  8. model.summary()
  9. #加载一张图像
  10. img=image.load_img(path='./MTB/Tuberculosis/Tuberculosis-203.png',target_size=(img_width,img_height))
  11. img_tensor=img_to_array(img)
  12. img_tensor=img_tensor.reshape((1,)+img_tensor.shape)
  13. img_tensor/=255.
  14. plt.imshow(img_tensor[0])
  15. plt.show()
  16. M=25#提取前M层的输出
  17. layer_outputs=[layer.output for layer in model.layers[:M]]
  18. activation_model=Model(inputs=model.input,outputs=layer_outputs)
  19. #以预测模式运行模型 activations包含卷积层的M个输出
  20. activations=activation_model.predict(img_tensor)
  21. print(activations[M-1].shape)
  22. first_layer_activation = activations[M-1]
  23. L=3 # 显示第L个通道的图像
  24. plt.matshow(first_layer_activation[0, :, :, L-1], cmap='viridis')
  25. plt.show()

 输出如下:

 第24层输出的图像,黄色的部分是模型重点关注的区域!

我们再看看更深层次的图像,比如第80层:

 太过于抽象,哈哈哈哈!

(c)可视化过滤器(卷积核)

  1. # 获取基础模型的第一层,即Conv2D层的权重
  2. for layer in base_model.layers:
  3. if isinstance(layer, Conv2D):
  4. weights = layer.get_weights()[0]
  5. break
  6. # 将权重缩放到0-1
  7. weights_min = weights.min()
  8. weights_max = weights.max()
  9. weights = (weights - weights_min) / (weights_max - weights_min)
  10. # 获取过滤器数量
  11. num_filters = weights.shape[-1]
  12. # 循环遍历所有过滤器
  13. for i in range(num_filters):
  14. # 获取第i个过滤器
  15. filter = weights[:, :, :, i]
  16. # 为过滤器创建一个子图
  17. plt.subplot(num_filters//4 + 1, 4, i+1)
  18. # 显示过滤器
  19. plt.imshow(filter)
  20. # 删除坐标轴
  21. plt.axis('off')
  22. # 显示整个图
  23. plt.show()

这个更加抽象哈:

三、写在最后

激活映射是每个卷积层通过卷积操作与激活函数处理后的输出结果,它展示了网络在特定层级对输入图像的响应,可视化这些激活映射有助于我们理解模型学习到的特征与决策过程。通过上面的例子,相信大家对于CNN的原理有更直观的认识吧。

四、数据

链接:https://pan.baidu.com/s/15vSVhz1rQBtqNkNp2GQyVw?pwd=x3jf

提取码:x3jf

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/132993?site
推荐阅读
相关标签
  

闽ICP备14008679号