Image Inpainting with Learnable Bidirectional Attention Maps

Image Inpainting with Learnable Bidirectional Attention Maps

pytorch

引言

部分卷积（PConv）的缺陷：

1 将含有1个有效值像素的区域与含有9个有效值像素的区域同等对待

2 Partial convolution中，无效pixel将在深层中逐渐消失，从而将所有mask值逐渐转换为1。而且在Decoder部分不起作用。

3 PCONV只前向更新

基于PCONV的改进

 

 

 

网络框架

 

反向传播

 

 

损失函数