python - 将一个层从一个 CNN 模型复制到另一个。 (layer_from_config 在版本 2 中不起作用)

Question

在 Keras 2 中，layer_from_config 已从 keras.utils.layer_utils 中删除。有谁知道有什么替代品吗？

详细说明:我有一个经过训练的 CNN 模型。我需要将一个图层从该模型复制到另一个模型。早些时候我曾经做过layer_from_config和set_weights函数。但它们在 Keras 2.0 中被删除了。需要帮助来完成此功能。

谢谢

Answer 1

该函数现在是 Layers 类的类函数(这似乎更有意义)。 set_weights 也是如此。使用方法(the doc is up to date) :

layer = Dense(32)
config = layer.get_config()
reconstructed_layer = Dense.from_config(config)

所以你需要知道你想要重建的图层的类名。或者您可以构建一个如下所示的字典，其中包含类名(以便您可以将配置存储在某处以在空代码中重建图层):

from keras import layers

config = layer.get_config()
layer = layers.deserialize({'class_name':      layer.__class__.__name__,
                        'config': config})

有帮助吗？