繁体   English   中英

在Keras顺序模型中,如何用我自己的权重初始化第二个conv层?

[英]How to Initialize with my own weights for the second conv layer in a Keras sequential model?

我有一个具有两个卷积层的模型。 我已经成功为conv_1图层设置了新的权重,但是在设置conv_2图层的权重conv_2收到错误消息:

    model.add(Conv2D(8, (3, 3), input_shape=(28,28,1), activation='relu'))
    model.add(MaxPooling2D(pool_size=(2, 2)))
    model.add(Conv2D(6, (3, 3), input_shape=(26,26,1), activation='relu'))

    model.layers[0].set_weights(w1)
    model.layers[2].set_weights(w2)

在这里, w1.shape == (3, 3, 1, 8)w2.shape == (3, 3, 1, 6) 错误消息是:

ValueError: Layer weight shape (3, 3, 8, 6) not compatible with provided weight shape (3, 3, 1, 6)

我不明白为什么不设置权重?

正如我在评论部分中提到的,一种替代方法是对过滤器中的所有通道使用相同的权重。 为此,您可以轻松地将w2的值重复八次,以获得形状为(3,3,8,6)的数组:

w2 = w2.repeat(8,axis=2)
w2.shape
# (3,3,8,6)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM