Imported from https://github.com/DAA233/generative-inpainting-pytorch