Skip to main content
 首页 » 编程设计

python之有些人可以使用相同的内核参数重复使用tensorflow或keras中的Con2d吗

2024年09月03日27lexus

我刚刚遇到一个奇怪的想法,要求我重复使用同一个卷积层。顺便说一句,我正在使用 tensorflow 1.13.1 和 keras 2.2.4。

在 keras 的通常情况下,它就像,

x = Conv2D()(x) 
x = Conv2D()(x) 
... 

但这意味着应用了 2 个不同的 2D 卷积层,我想重复使用相同的卷积核。

我发现了 tf.contrib.layers.repeat这似乎回答了我的问题。

有没有其他方法可以做同样的事情?或者根本不值得尝试。谢谢!!!

请您参考如下方法:

您需要为此使用 Keras Functional API:

layer = Conv2D(...) 
input = Input(...) 
x = input 
 
x = layer(x) 
x = layer(x) 

这两次应用完全相同的卷积层,具有相同的权重/内核。