如何创建过滤器不在一个维度上移动的转换图层

时间:2017-11-22 21:20:21

标签: python tensorflow convolution

换句话说,我希望过滤器在一个维度上覆盖图像,只在另一个方向上移动。

如果我有一个带有形状的输入(忽略批量大小和输入通道数)(h, w),我希望有一个过滤器,其形状为(x, w) x<h。创建一个形状为(x, w)的过滤器并使用'VALID'的填充是正确的事情(VALID的想法会强制过滤器仅在过滤器时在图像上进行卷积'适合'宽度尺寸中的图像'?

可能还有另一种更好的方法吗?

1 个答案:

答案 0 :(得分:2)

是的,你可以这样做。另一种方法是使用stride参数指定每个维度的步幅。您可以将其他维度的步幅设置为0.我认为stride=(1, 0)是正确的语法。