Partial Convolution implement Tensorflow 1.X version
inpainting里面重要的一个工作是Nvidia的Partial Convolution部分卷积,该卷积包含两个分支,一个是更新mask图像,另一个则是根据mask图进行部分的卷积。
下面是他的1.x版本的实现代码如下:
def pc_conv(x, mask, cnum, ksize, stride, test ,name='pconv'):
b, h, w, c = int(x.s...
原创
2020-03-09 14:03:28 ·
552 阅读 ·
0 评论