将图层从一个CNN模型复制到另一个。(来自配置的层在版本2中不工作)

2024-05-18 06:33:04 发布

您现在位置:Python中文网/ 问答频道 /正文

在Keras 2中,来自配置的层从中移除keras.utils.layer\u utils。有人知道有什么替代品吗。在

详细说明: 我有一个受过训练的CNN模特。我需要从那个模型复制一个图层到另一个图层。之前我用了从配置层和设置权重函数。但它们在keras2.0中被删除了。需要帮助才能完成此功能。在

谢谢


Tags: 函数模型功能图层layer替代品utilscnn
1条回答
网友
1楼 · 发布于 2024-05-18 06:33:04

该函数现在是类层的类函数(这似乎更有意义)。设定重量相同。 使用方法(the doc is up to date)

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

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

^{pr2}$

有帮助吗?在

相关问题 更多 >