如何在keras中实现膨胀卷积?

时间:2019-02-21 01:37:47

标签: python tensorflow keras

我想在Keras中使用膨胀卷积。我找到了AtrousConv2D,但在Keras文档中以及使用时都找不到它的任何定义

acov=AtrousConv2D((3,3))(image)

它会产生此错误

  

init ()缺少1个必需的位置参数:“ kernel_size”

我需要膨胀卷积,但是我不知道该如何使用该层或如何自己生成该层。

1 个答案:

答案 0 :(得分:2)

标准keras Conv2D层支持膨胀,您只需要将dilation_rate设置为大于1的值即可。例如:

out = Conv2D(10, (3, 3), dilation_rate=2)(input_tensor)