Improvement of Bayer-pattern Demosaicking with Dictionary Learning Algorithm
Zhu Bo①② Wen De-sheng① Wang Fei③ Li Hua Song Zong-xi①
①(Xi’an Institute of Optics and Precision Mechanics, Chinese Academy of Sciences, Xi’an 710119, China)
②(Graduate University of Chinese Academy of Sciences, Beijing 100049, China)
③(Xi’an Jiaotong University, Xi’an 710049, China)
④(Shangluo University, Shangluo 726000, China)
Abstract:Demosaicking is important for the quality of digital images in resource-constrained single chip devices. This paper presents an improved dictionary learning-based color demosaicking algorithm. Firstly, an initial interpolation is applied to the,channel by Local Directional Interpolation (LDI) and fused by analysis the joint distribution of the gradient. Gaussian Mixture Model (GMM)-based clustering is used to classify dictionary image into different classes. The Principal Component Analysis (PCA) is performed on these classes to choose the principal components for the dictionary construction. And then, dictionary learning is applied to obtain the interpolated G^ and the lost R^ and B^ are interpolated by the help of the reconstructed G^, accordingly. Since, R^, G^ and B^ of the given pixels are better represented, the whole image can be reconstructed accurately. Taking McMaster color image dataset as dictionary, standard image and image from DALSA CMOS camera are used for effect evaluation of the demosaicking algorithm. Experimental results prove that the proposed algorithm outperforms some state-of-the-art demosaicking methods both in PSNR measure and visual quality.