Tensorflow中的图像修复模型实现
“使用部分卷积对不规则Kong进行图像修补”第一个tensorflow主实例,使用tensorflow完全实现,而无需修改源代码。
原始论文:
YouTube上的演示:
部分卷积
使用curr_bin_mask表示当前二进制文件的掩码; conved_mask表示二进制掩码的卷积结果,对应于文本中的sum(M); new_bin_mask表示卷积后的新二进制掩码,更新规则为:
((conved_mask==0)==0)
因此,局部卷积的计算如下:
Pconv(x) = (Conv(x*curr_bin_mask)*conved_mask+b)*new_bin_mask
如文本中所述,使用新掩码的操作是为了确保无效输入为零。
网络结构
用局部卷积代替卷积
面膜生成
与原始版本不同,我使用opencv生成了一个掩码,并将无效零件输入设置为零。
2022-02-25 17:23:58
86KB
系统开源
1