Rain removal via shrinkage of sparse codes and learned rain dictionary

Chang-Hwan Son,Xiao-Ping Zhang
DOI: https://doi.org/10.1109/ICMEW.2016.7574752
2016-01-01
Abstract:Recently, sparse coding and dictionary learning have been widely used for feature learning and image processing. They can also be applied to the rain removal by learning two types of rain and non-rain dictionaries, and then forcing the sparse codes of the rain dictionary to be zero vectors. However, this approach can generate edge artifacts that appear in the non-rain regions, especially around the edges of objects. Based on this observation, a new approach of shrinking the sparse codes is presented in the paper. To effectively shrink the sparse codes in the rain and non-rain regions, an error map between the input rain image and the reconstructed rain image with the learned rain dictionary is generated. Based on this error map, the sparse codes of the rain and non-rain dictionaries are used together to represent the image structures of objects and avoid the edge artifacts in the non-rain regions. In the rain regions, the correlation matrix between the rain and non-rain dictionaries is calculated and then the sparse codes corresponding to the highly correlated signal-atoms between the rain and non-rain dictionaries are shrunk together to improve the removal of the rain structures. The experimental results show that the proposed approach using the shrinkage of the sparse codes can preserve image structures and avoid the edge artifacts in the non-rain regions, while removing the rain structures in the rain regions.
What problem does this paper attempt to address?