将图层从一个CNN模型复制到另一个。 (layer_from_config在ver 2中不起作用)

时间:2017-03-16 22:25:21

标签: python tensorflow keras keras-layer

在Keras 2中, layer_from_config 已从 keras.utils.layer_utils 中删除。有谁知道任何替代品。

详细说明: 我有一个训练有素的CNN模型。我需要将该模型中的图层复制到另一个模型。之前我曾经做过layer_from_config和set_weights函数。但它们在Keras 2.0中删除了。需要帮助才能执行此功能。

由于

1 个答案:

答案 0 :(得分:2)

该函数现在是类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})

有帮助吗?