Image Inpainting Using MSCSWin Transformer and Color Correction

Abstract

Image inpainting has been researched for many years. From traditional methods to current CNN models, they all pursue two targets (structural stability and texture consistency). In this paper, we propose the multi-shift CSWin Transformer (MSCSWin Transformer) and the HSV loss to focus on colors to inpaint images for these two targets. At last, we compare our model with state-of-the-art methods on the Places2 dataset to confirm our proposed module is indeed working.