Citation: | Wang Fei, Wang Wei. An automatic white balance method via dark channel prior[J]. Opto-Electronic Engineering, 2018, 45(1): 170549. doi: 10.12086/oee.2018.170549 |
[1] | Lukac R. Refined automatic white balancing[J]. Electronics Letters, 2007, 43 (8): 445-446. doi: 10.1049/el:20070142 |
[2] |
Zhang B Y, Batur A U. A real-time auto white balance algorithm for mobile phone cameras[C]//Proceedings of 2012 IEEE International Conference on Consumer Electronics, 2012: 1-4. |
[3] | Buchsbaum G. A spatial processor model for object colour perception[J]. Journal of the Franklin Institute, 1980, 310 (1): 1-26. doi: 10.1016/0016-0032(80)90058-7 |
[4] | Liu Y C, Chan W H, Chen Y Q. Automatic white balance for digital still camera[J]. IEEE Transactions on Consumer Electronics, 1995, 41 (3): 460-466. doi: 10.1109/30.468045 |
[5] |
Weng C C, Chen N H, Fuh C S. A novel automatic white balance method for digital still cameras[C]//Proceedings of IEEE International Symposium on Circuits and System, 2005, 4: 3801-3804. |
[6] | Dong C, Loy C C, He Kaiming, et al. Learning a deep convolutional network for image super-resolution[M]//FLEET D, PAJDLA T, SCHIELE B, et al. Computer Vision-ECCV 2014. Cham: Springer, 2014: 184-199. |
[7] | Cheng D L, Price B, Cohen S, et al. Effective learning-based illuminant estimation using simple features[C]//Proceedings of 2015 IEEE Conference on Computer Vision and Pattern Recognition, 2015: 1000-1008. |
[8] | Barron J T. Convolutional color constancy[C]//Proceedings of 2015 IEEE International Conference on Computer Vision, 2015: 379-387. |
[9] | Barron J T. Convolutional color constancy[C]//Proceedings of 2015 IEEE International Conference on Computer Vision, 2015: 379-387. |
[10] | Liu Y C, Chan W H, Chen Y Q. Automatic white balance for digital still camera[J]. IEEE Transactions on Consumer Electronics, 1995, 41(3): 460-466. doi: 10.1109/30.468045 |
[11] | He K M, Sun J, Tang X O. Single image haze removal using dark channel prior[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011, 33(12): 2341-2353. doi: 10.1109/TPAMI.2010.168 |
[12] | Yeh C H, Kang L W, Lee M S, et al. Haze effect removal from image via haze density estimation in optical model[J]. Optics Express, 2013, 21(22): 27127-27141. doi: 10.1364/OE.21.027127 |
[13] |
Gehler P V, Rother C, Blake A, et al. Bayesian color constancy revisited[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition, 2008: 1-8. |
[14] | Cheng D L, Prasad D K, Brown M S. Illuminant estimation for color constancy: why spatial-domain methods work and the role of the color distribution[J]. Journal of the Optical Society of America A, 2014, 31 (5): 1049-1058. doi: 10.1364/JOSAA.31.001049 |
Overview: Color is an important feature in the field of computer vision, as it relates to the practical vision problems such as designing an object recognition or classification tasks. For human beings, the brain will let people feel who is in a natural white light environment, while this is difficult for a computer, Given a pixel with blue color, how can a computer distinguish that the color is resulted by a white object under a blue lighting source or a blue object under a white lighting source? In order to address this question, we need to remove illuminated color of the light source, which is specified as white balance problem. In general, there are three types of color constancy approaches, summarized as statistics-based, traditional learning based and deep learning based ones, respectively. Although the latter two have achieved excellent results, but due to the low speed, it is not practical. The statistics-based method has the problem of white region or white points detection error which leads to white balance failure. In order to solve this problem this paper proposes a white balance method based on dark channel prior. First, get the dark channel image, extract the white region in the image according to the dark channel, and then remove the region with high saturation. Finally, in order to correct the color and ensure that the image brightness does not change, we calculate the correction gain in the CIE-XYZ color space relative to the luminance channel Y. By removing the region with high saturation, we define a threshold transformation. Through a large number of experiments, we obtain a threshold K which can effectively eliminate the high light area, and this makes the white area more stable. In order to make the algorithm run on a low frequency ARM, we tested the white balance effect under different down sampling which shows that the speed is only 5ms under the size of 1/16, and the effect does not have much impact. Experimental results show that our algorithm has achieved good results both in subjective and objective evaluation compared with some classical algorithms. Meanwhile, we compared our method with a Nikon camera. Our method is excellent, and it's better than Canon in detail. We use our algorithm instead of the white balance algorithm in HI3516D. The test shows that our method can achieve 150 frame/sec and the effect is better than the algorithm in HI3516D.
White region extraction process. (a) Original color cast image. (b) Dark channel image. (c) Light transmission model; (d) The final stable white region. (The region of the red frame from the visual point of view is very obvious in the supersaturated region, and our algorithm effectively eliminates this part of the region when extracting white regions.)
Error comparison of different K values
White balance results of the indoor image using five different automatic white balance methods. (a) Original image (53.95); (b) GWM (26.92); (c) PRM (27.02); (d) Grey-Edge (19.11); (e) Ada-Threshold (22.80); (f) Our method(11.91).
White balance results of the outdoor image using five different automatic white balance methods. (a) Original image (59.03); (b) GWM (11.40); (c) PRM (11.70); (d) Grey-Edge (5.02); (e) Ada-Threshold (8.06); (f) Our method (4.12)
White balance results using Nikon D7100 and our method. (a), (d) Original image; (b), (e) D7100 auto white balance; (c), (f) Our auto white balance
White balance results using HI3516D and our method. (a), (d) Original image; (b), (e) HI3516D auto white balance; (c), (f) Our auto white balance