Image rain removing method based on attention mechanism and gating circulation unit

Abstract

The invention provides an image rain removal method based on an attention mechanism and a gating circulation unit, and the method comprises the steps: firstly constructing an image rain removal network architecture based on the attention mechanism and the gating circulation unit, wherein the image rain removal network architecture includes 6 modules, each of the first five modules comprises a gating circulation unit, a spatial attention module and an activation function; wherein the module 1 is used as an encoder, expansion convolution with expansion factors of 1, 2, 4 and 8 is respectively used in the module 2 to the module 5, and the sizes of corresponding receiving domains are respectively 5 * 5, 9 * 9, 17 * 17 and 33 * 33; the module 6 comprises a convolution layer, a channel attentionmodule and an activation function, and a 1 * 1 convolution layer is connected behind the module 6 and serves as a decoder to generate residual error mapping; then, selecting the MSE and the SSIM as loss functions; and finally, training the constructed network architecture. According to the method, the problems of rain stripe residues and detail blurring occurring in processing of the image containing the dense rain stripes are solved, the rain lines in the image are removed, meanwhile, the detail parts of the image are reserved, and the image definition is greatly improved.

Description

technical field [0001] The invention belongs to the field of image processing, and in particular relates to an image rain removal method based on an attention mechanism and a gated cycle unit. Background technique [0002] Rain is a relatively common dynamic severe weather. The raindrops fall randomly, the speed of falling is very fast, and they are randomly distributed in the air. In the image, the raindrops mainly appear in the shape of rain lines, which is easy to cause blurred images, loss of detail information, and even Some areas in the image will be randomly occluded, which greatly affects the visual effect of the image and also reduces the performance of computer vision algorithms (such as person detection and tracking, automatic driving, traffic monitoring, etc.). For example, when an image is taken in a rainy environment, due to the influence of the rain line, the characteristics of the object in the image will change, which will reduce the accuracy of the target d...

AI Technical Summary

Problems solved by technology

[0004] Purpose of the invention: In order to overcome the deficiencies of the prior art, the present invention provides an image deraining method based on the attention mechanism and the gated cycle unit, which solves the problems of reduced image contrast, blurred details and residual rain lines when the image is derained

Images