0
self.conv6 = tf.keras.layers.Conv2D(1024, kernel_size=3, padding=6,dilation_rate=6) # atrous convolution

这在 tensorflow 2.0 中出现错误
我看到了 padding='same' 选项
但是如果我想给 padding=6 怎么办

文件“/home/jake/venv/lib/python3.7/site-packages/tensorflow_core/python/keras/utils/conv_utils.py”,第 202 行,在 normalize_padding padding = value.lower() AttributeError: 'int' object没有“下”属性

4

1 回答 1

0

根据tf.keras.layers.Conv2D文档,padding只能是'same''valid'

'same'确保保留空间维度并'valid'添加确保过滤器“适合”您的空间维度所需的最小填充量(如果stride不是一个,这可能是一个问题)。

您不需要指定自定义填充大小,因为在大多数情况下,这将是不必要的上采样。但是如果你必须,你可以在 conv 层之前有一个 padding 层。

tf.keras.layers.ZeroPadding2D(padding=(6, 6)) # put this before your conv layer
于 2020-05-31T07:25:34.020 回答