tensorflow - keras自定义conv2d初始化

Question

我一直在尝试弄清楚如何在 Keras 中为 Conv2D 层创建自定义初始化。任何人都可以举例说明如何手动创建“过滤器”或权重模型作为训练的起点。即我想将一个带有权重的二维矩阵传递给 Conv2D 层，该矩阵可以用作起点。矩阵可以是高斯模糊滤波器或锐化滤波器或边缘检测器等。

小红书

Answer 1

来自 custom initializers 上的 Keras 文档:

from keras import backend as K

def my_init(shape, dtype=None):
    return K.random_normal(shape, dtype=dtype)    

model.add(Dense(64, init=my_init))

因此，您可以在 my_init() 中轻松定义适当的自定义初始化，并在您的 Conv2D 层中使用它。