Partial Convolution based Padding
Guilin Liu, Kevin J. Shih, Ting-Chun Wang, Fitsum A. Reda, Karan Sapra, Zhiding Yu, Andrew Tao, Bryan Catanzaro
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/NVIDIA/partialconvOfficialIn paperpytorch★ 0
- github.com/feixuetuba/inpaintingpytorch★ 0
- github.com/feixuetuba/inpatingpytorch★ 0
- github.com/lessw2020/auto-adaptive-aipytorch★ 0
Abstract
In this paper, we present a simple yet effective padding scheme that can be used as a drop-in module for existing convolutional neural networks. We call it partial convolution based padding, with the intuition that the padded region can be treated as holes and the original input as non-holes. Specifically, during the convolution operation, the convolution results are re-weighted near image borders based on the ratios between the padded area and the convolution sliding window area. Extensive experiments with various deep network models on ImageNet classification and semantic segmentation demonstrate that the proposed padding scheme consistently outperforms standard zero padding with better accuracy.