为Keras模型的给定输入打印所有尺寸的形状值

import keras
from keras.models import Model

model = keras.models.load_model('model.h5')  # 加载已保存的模型
# 或者
# model = ...  # 创建新的模型

layer_outputs = [layer.output for layer in model.layers]  # 获取每个层的输出
activation_model = Model(inputs=model.input, outputs=layer_outputs)  # 创建新的模型

import numpy as np
input_data = np.random.random((1, 224, 224, 3))  # 准备输入数据，这里假设输入数据的形状是(1, 224, 224, 3)

activations = activation_model.predict(input_data)  # 预测每个层的输出
for layer_activation in activations:
    print(layer_activation.shape)  # 打印每个层的形状