【问题标题】:how to give padding=6 in tensorflow 2.0?如何在 tensorflow 2.0 中给出 padding=6?
【发布时间】:2020-09-18 13:37:15
【问题描述】:
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 has no attribute 'lower'

【问题讨论】:

    标签: python tensorflow tensorflow2.0 tensorflow2.x


    【解决方案1】:

    根据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
    

    【讨论】:

      猜你喜欢
      • 2019-09-13
      • 1970-01-01
      • 2019-09-09
      • 2019-09-05
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2019-09-19
      • 1970-01-01
      相关资源
      最近更新 更多