Rain streak removal using learnable hybrid MAP network
Long Ma,Risheng Liu,Zhiying Jiang,Yiyang Wang,Xin Fan,Haojie Li
DOI: https://doi.org/10.11834/jig.170390
2018-01-01
Journal of Image and Graphics
Abstract:Objective In recent years,artificial intelligence has become the emerging area and attract the attention of many research institutions.Computer vision is the force at the core of the development in the area of artificial intelligence.Image restoration remains an active research topic in low-level computer vision.Because the distribution of the rain streaks is irregular and the problem is ill-posed,rain removal for the single image is one of the difficult problems to be solved.So far,the basic idea to solve the problem is building the rational rain image generative model,and find the solution by different measures.There are two representative ways.One is by describing the priors of rain streaks and background with conven tional models and then solve it by the optimization algorithm,but the shortcoming is due to the interdependent connection between rain streaks and the background,it is hard to balance the relationship between the performance of rain removal and the clarity of background.The other is deep neural network based framework to build the network structure and loss function,then learning the network with large amounts of data,but the shortcoming is the difficulty of generalization because of test results are subject to the training data which is hard to cover the rain in the different scene.Method For the shortcomings mentioned above,inspired by general image restoration,this paper combines the network by data-driven with the empirical model,and the process displays the advantages of the network and empirical model,then learnable mixture MAP network is proposed so that image deraining problem can be solved effectively.The specific process is to build the energy model which contains the implicit prior item based on maximum a posterior (MAP) framework,then transfer the model to two parts that is estimating background and estimating rain streaks by the optimization algorithm,and that reduce the dependency of background and rain streaks.For the first part,it can be solved by the residual network for denoising by analysis the model and the optimization objective and in this way,the estimated image is more clearly.For the second part,the rain streaks prior to the input image can be described real-time by gaussian mixture model to improve the accuracy of estimating rain streaks.Result In the end,compared with the three empirical model-based algorithms and two deep learningbased algorithms,the experiments on the synthetic dataset and real-world image demonstrate the good performance of our model on removing rain streaks meanwhile reducing the information loss of the background,in addition,the SSIM value reaches to 0.92 on the synthetic dataset.Conclsion This paper proposes to combine model-based method with deep network-based method,assimilates the advantages from each other.Not only can the majority of visible rain streaks be re moved,but also remains detail information of background.Achieving a harmonious of the two aspects and standing out from state-of-the-art methods.At the same time,this paper makes a verification,which is the effective way to solve the image restoration problem by combining the conventional model and deep network.