如何在tensorflow 2.0中给予padding=6?

5t7ly7z5  于 2023-10-23  发布在  其他
关注(0)|答案(1)|浏览(133)

在下面的conv层中,如何为输入Tensor添加自定义填充大小?

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”对象没有属性“lower”

w1jd8yoj

w1jd8yoj1#

根据tf.keras.layers.Conv2D文档,padding只能是'same''valid'
'same'确保保留空间维度,'valid'添加所需的最小填充量,以确保过滤器“适合”您的空间维度(如果stride不是1,这可能是一个问题)。
您不需要指定自定义填充大小,因为在大多数情况下,这将是不必要的上采样。但是如果你必须的话,你可以在conv层之前有一个padding层。

tf.keras.layers.ZeroPadding2D(padding=(6, 6)) # put this before your conv layer

相关问题