用1s代替零的Tensorflow填充

Question

Tensorflow函數似乎只使用零填充（例如，tf.nn.conv2d函數）。

是否有一種簡單的方法（只需要對現有代碼進行最小限度的更改）就可以使用填充爲1或任意其他任意數字而不是零來運行相同的函數？

Answer 1

最簡單的方法是將conv2d中的填充從「SAME」更改爲「VALID」並自己執行填充。

img = tf.ones([img_height+padding,img_width+padding, channels]) 
img[padding//2:img_height, padding//2:img_width] = old_img 
result = conv2d(img, kernel, padding="VALID")