Rain image rain stripe removing method and system based on image filtering and CNN

Abstract

The invention provides a rain image rain stripe removing method and system based on image filtering and CNN. According to the method, a rain image data set is expanded and trained by using a rain image data set enhancement method, multiple groups of rain images and label images are acquired, the rain images are subjected to image filtering to obtain rain image high-frequency parts, the rain imagehigh-frequency parts are input into a rain removal network to obtain rain removal images, and an SSIM loss function is added to optimize the rain removal network; the rain removal image is input intothe H-G discrimination network to obtain a discrimination result, the discrimination result is fed back to the rain removal network to further improve the rain removal quality, and the problems that an existing rain removal method is difficult to accurately describe a rain model, rain stripes cannot be completely removed, and details are easy to lose are solved.

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