Rainwater-Removal Image Conversion Learning with Training Pair Augmentation

Entropy (Basel). 2023 Jan 6;25(1):118. doi: 10.3390/e25010118.

Abstract

In this study, we proposed an image conversion method that efficiently removes raindrops on a camera lens from an image using a deep learning technique. The proposed method effectively presents a raindrop-removed image using the Pix2pix generative adversarial network (GAN) model, which can understand the characteristics of two images in terms of newly formed images of different domains. The learning method based on the captured image has the disadvantage that a large amount of data is required for learning and that unnecessary noise is generated owing to the nature of the learning model. In particular, obtaining sufficient original and raindrops images is the most important aspect of learning. Therefore, we proposed a method that efficiently obtains learning data by generating virtual water-drop image data and effectively identifying it using a convolutional neural network (CNN).

Keywords: GAN; Pix2pix; augmentation learning; image-to-image learning; rainwater removal.