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