base_model = ResNet50(weights="imagenet", include_top=False, input_shape=(224, 224, 3))
truncated_model = Model(inputs = base_model.layers[7].input, outputs = base_model.layers[-1].output) #truncates the functional model from layer 7 to final layer
keras.utils.plot_model(truncated_model, "mini_resnet.png", show_shapes=True) #Plots the functional model graph
1条答案
按热度按时间wxclj1h51#
如果有人在未来看到这个问题，我在无数次谷歌搜索和多个论坛发帖后终于找到了答案。我在任何地方都没有找到答案，只是在函数中玩，直到我找到了解决方案。