Image defogging and rain-removing algorithm based on selective attention mechanism

Abstract

The invention discloses an image defogging and rain-removing algorithm based on a selective attention mechanism. The image defogging and rain-removing algorithm specifically comprises the following three processes of expanding a training sample, training a network on a training set, defogging and removing rain from a test image, and evaluating a restored image and a clear image. The method comprises s that: firstly, a defogging data set conducts cutting, intercepting, rotating and other operations on a used training set to expand an original data set; the pictures are input into a convolutional network based on a selective attention mechanism, and the network is trained by using a multi-loss function; and finally, the test pictures are input into the network to obtain defogging / rain-removing results. According to the invention, an end-to-end defogging / rain removing algorithm is realized, and rain-removing / defogging can be efficiently and accurately carried out on an image with rain / fog.

Description

technical field [0001] The invention relates to the technical field of image restoration, in particular to an image defogging and rain removal algorithm based on a selective attention mechanism. Background technique [0002] Rain and smog are becoming more and more common in our daily life. Images captured in hazy and rainy weather often have relatively low visibility and contrast. However, various vision tasks, such as image classification, semantic segmentation, and object detection, require sharp images as input. Smog and rain bands make these tasks impossible to complete well. Since the direction and density of rain bands in rainfall images and the depth of fog and atmospheric illumination in foggy images are inhomogeneous, image dehazing and denoising are ill-posed problems. In view of this, the research on image defogging and rain removal is of great significance. [0003] Most of the traditional fog and rain removal algorithms restore the final clear image based o...

AI Technical Summary

Problems solved by technology

[0004] In view of the existing image defogging and rain removal methods, which rely on the physical model and the restoration effect is not good enough, the present invention provides an image defogging and rain removal algorithm based on a selective attention mechanism.

Images