Keras中间层输出的两种方式,即特征图可视化

2022-12-29,,,,

训练好的模型,想要输入中间层特征图,有两种方式:

1. 通过model.get_layer的方式。创建新的模型,输出为你要的层的名字。

创建模型,debug状态可以看到模型中,base_model/layers,图中红框即为layer名字,根据你想输出的层填写。最后网络feed数据后,输出的就是中间层结果。

2. 通过建立Keras的函数。

 from keras import backend as K
from keras.models import load_model
from matplotlib import pyplot as plt
import cv2
import numpy as np def main():
model = load_model('../Project/weights.best_10-0.90.hdf5') images=cv2.imread("../Project/1.jpg")
# cv2.imshow("Image", images)
cv2.waitKey(0) # Turn the image into an array.
# 根据载入的训练好的模型的配置,将图像统一尺寸
image_arr = cv2.resize(images, (70, 70)) image_arr = np.expand_dims(image_arr, axis=0) # 第一个 model.layers[0],不修改,表示输入数据;
# 第二个model.layers[ ],修改为需要输出的层数的编号[]
layer_1 = K.function([model.layers[0].input], [model.layers[1].output]) # 只修改inpu_image
f1 = layer_1([image_arr])[0] # 第一层卷积后的特征图展示,输出是(1,66,66,32),(样本个数,特征图尺寸长,特征图尺寸宽,特征图个数)
for _ in range(16):
show_img = f1[:, :, :, _]
show_img.shape = [66, 66]
plt.subplot(4, 4, _ + 1)
# plt.imshow(show_img, cmap='black')
plt.imshow(show_img, cmap='gray')
plt.axis('off')
plt.show() if __name__ == '__main__':
main()

特征图可视化结果:

Keras中间层输出的两种方式,即特征图可视化的相关教程结束。