Method and system for removing rain streaks in rainy images based on image filtering and CNN

Abstract

The present invention provides a method and system for removing rain stripes from rainy images based on image filtering and CNN. The present invention uses a rain image data set enhancement method to expand the training rain image data set to obtain multiple sets of rainy images and label images. The rainy images pass through The high-frequency part of the rainy image is obtained by image filtering, and the high-frequency part of the rainy image is input into the rain-removing network to obtain the rain-removing image, and the SSIM loss function is added to optimize the rain-removing network; the rain-removing image is input into the H-G discriminant network to obtain the discrimination result. The discrimination results are fed back to the deraining network to further improve the quality of deraining, and solve the problems that the existing deraining methods are difficult to accurately describe the rain model, the rain streaks cannot be removed cleanly, and the details are easily lost.

Description

technical field [0001] The invention relates to the field of image processing, in particular to a method and system for removing rain streaks from rainy images based on image filtering and CNN (convolutional neural network). Background technique [0002] Visual image systems have important applications in the field of computer vision. However, most visual image processing systems only consider the ideal situation of indoor or good weather, and do not consider the impact of the actual environment. The images collected by the visual image system in rainy days often contain rain streaks. These rain streaks will affect the imaging quality of the image, resulting in a decrease in the algorithm accuracy of visual image systems such as target detection and dam monitoring. Therefore, in order to improve the imaging quality of the visual image system in rainy days, it is of great significance to study the image deraining. [0003] In the real visual sensing image system, there are ...

AI Technical Summary

Problems solved by technology

[0005] The technical problem to be solved by the present invention is that it is still difficult to accurately describe the rain model in the existing single-image rain removal method mentioned above, and it is difficult to separate the background image from the rain streaks, so that the rain streaks cannot be removed. For the technical problem of high algorithm complexity, a method and system for removing rain streaks in rainy images based on image filtering and CNN are provided to solve the above technical defects

Images