Image in-painting for irregular holes using partial convolutions
AssigneeNVIDIA Corporation
InventorsGuilin Liu, Fitsum A. Reda, Kevin Shih, Ting-Chun Wang, Andrew Tao, Bryan Catanzaro
Life of this patent
Estimating the term…
Loading patent…
AssigneeNVIDIA Corporation
InventorsGuilin Liu, Fitsum A. Reda, Kevin Shih, Ting-Chun Wang, Andrew Tao, Bryan Catanzaro
Estimating the term…
A neural network architecture is disclosed for performing image in-painting using partial convolution operations. The neural network processes an image and a corresponding mask that identifies holes in the image utilizing partial convolution operations, where the mask is used by the partial convolution operation to zero out coefficients of the convolution kernel corresponding to invalid pixel data for the holes. The mask is updated after each partial convolution operation is performed in an encoder section of the neural network. In one embodiment, the neural network is implemented using an encoder-decoder framework with skip links to forward representations of the features at different sections of the encoder to corresponding sections of the decoder.
A computer-implemented method for performing an image in-painting operation, the method comprising:
A system, comprising:
A non-transitory computer-readable media storing computer instructions for performing image in-painting that, when executed by one or more processors, cause the one or more processors to perform the steps of:
Dependencies are inferred from the claim text.