Improved Zero-DCE Low-Light Enhancement Algorithm for Night Fatigue Driving Detection
HUANG Zhenyu1,2, CHEN Yutao1,2,3, LIN Dingci1,2, HUANG Jie1,2,3
1. College of Electrical Engineering and Automation, Fuzhou University, Fuzhou 350108; 2. 5G+ Industrial Internet Institute, Fuzhou University, Fuzhou 350108; 3. Key Laboratory of Industrial Automation Control Technology and Information Processing of Fujian Province, Fuzhou University, Fuzhou 350108
摘要 为了提高夜间疲劳驾驶检测的准确率,在现有低光增强算法Zero-DCE(Zero-Reference Deep Curve Estimation)的基础上,提出改进Zero-DCE的低光增强算法.首先,引入上下采样结构,减少噪声影响.同时,引入注意力门控机制,提高网络对图像中人脸区域的敏感性,有效提高网络的检测率.然后,针对噪声相关问题,提出改进的核选择模块.进一步,使用MobileNet的深度可分离卷积替换Zero-DCE的标准卷积,提高网络的检测速度.最后,通过人脸关键点检测网络和分类网络,判断驾驶员的疲劳状态.实验表明,在夜间环境下,相比现有的疲劳驾驶检测算法,文中算法在人脸检测的准确率和眼睛状态的识别率上都有所提升,取得较令人满意的检测效果.
Abstract:Grounded on the existing low-light enhancement algorithm, zero-reference deep curve estimation(Zero-DCE), an improved Zero-DCE low-light enhancement algorithm is proposed to increase the accuracy of night fatigue driving detection. Firstly, the upper and lower sampling structure is introduced to reduce the influence of noise. Secondly, the attention gating mechanism is employed to improve the sensitivity of the network to the face region in the image, and thus the detection rate is increased effectively. Then, an improved kernel selecting module is proposed for the problems arising from noise. Furthermore, standard convolution of Zero-DCE is replaced by the depthwise separable convolution of MobileNet to accelerate the detection. Finally, the driver fatigue state can be judged by the face key point detection network and classification network. The experimental results show that the proposed algorithm improves the accuracies of face detection and eye state recognition rate in a night environment with satisfactory detection results compared with the existing fatigue driving detection algorithms.
[1] KARTSCH V J, BENATTI S, SCHIAVONE P D, et al. A Sensor Fusion Approach for Drowsiness Detection in Wearable Ultralow-Power Systems. Information Fusion, 2018, 43: 66-76. [2] LI Y Y, YAMAMOTO T, ZHANG G N.The Effect of Fatigue Dri-ving on Injury Severity Considering the Endogeneity. Journal of Safety Research, 2018, 64: 11-19. [3] SIKANDER G, ANWAR S.Driver Fatigue Detection Systems: A Review. IEEE Transactions on Intelligent Transportation Systems, 2019, 20(6): 2339-2352. [4] KRAJEWSKI J, SOMMER D, TRUTSCHEL U, et al. Steering Wheel Behavior Based Estimation of Fatigue// Proc of the 5th International Driving Symposium on Human Factors in Driver Assessment, Training and Vehicle Design. Berlin, Germany: Springer, 2009: 118-124. [5] MARDI Z, ASHTIANI S N M, MIKAILI M. EEG-Based Drowsiness Detection for Safe Driving Using Chaotic Features and Statistical Tests. Journal of Medical Signals and Sensors, 2011, 1(2): 130-137. [6] WANG M, GUO L, CHEN W Y.Blink Detection Using Adaboost and Contour Circle for Fatigue Recognition. Computers and Electrical Engineering, 2017, 58: 502-512. [7] 朱虎明,李佩,焦李成,等.深度神经网络并行化研究综述.计算机学报, 2018, 41(8): 1861-1881. (ZHU H M, LI P, JIAO L C, et al. Review of Parallel Deep Neural Network. Chinese Journal of Computers, 2018, 41(8): 1861-1881.) [8] IBRAHIM H, KONG N S P. Brightness Preserving Dynamic Histogram Equalization for Image Contrast Enhancement. IEEE Transactions on Consumer Electronics, 2007, 53(4): 1752-1758. [9] WANG Q, WARD R K.Fast Image/Video Contrast Enhancement Based on Weighted Thresholded Histogram Equalization. IEEE Transactions on Consumer Electronics, 2007, 53(2): 757-764. [10] LAND E H.The Retinex Theory of Color Vision. Scientific American, 1977, 237(6): 108-128. [11] GUO X J, LI Y, LING H B.LIME: Low-Light Image Enhancement via Illumination Map Estimation. IEEE Transactions on Image Processing, 2017, 26(2): 982-993. [12] LÜ F F, LU F, WU J H, et al. MBLLEN: Low-Light Image/Video Enhancement Using CNNs[C/OL]. [2022-04-25]. http://vigir.missouri.edu/~gdesouza/Research/Conference_CDs/BMVC2018/contents/papers/0700.pdf. [13] JOBSON D J, RAHMAN Z, WOODELL G A.Properties and Performance of a Center/Surround Retinex. IEEE Transactions on Image Processing, 1997, 6(3): 451-462. [14] WEI C, WANG W J, YANG W H, et al. Deep Retinex Decomposition for Low-Light Enhancement[C/OL].[2022-04-25]. https://arxiv.org/pdf/1808.04560.pdf. [15] ZHANG Y H, ZHANG J W, GUO X J.Kindling the Darkness: A Practical Low-Light Image Enhancer// Proc of the 27th ACM International Conference on Multimedia. New York, USA: ACM, 2019: 1632-1640. [16] 欧嘉敏,胡晓,杨佳信.改进Retinex-Net 的低光照图像增强算法.模式识别与人工智能, 2021, 34(1): 77-86. (OU J M, HU X, YANG J X.Low-Light Image Enhancement Algorithm Based on Improved Retinex-Net. Pattern Recognition and Artificial Intelligence, 2021, 34(1): 77-86.) [17] JIANG Y F, GONG X Y, LIU D, et al. EnlightenGAN: Deep Light Enhancement Without Paired Supervision. IEEE Transactions on Image Processing, 2021, 30: 2340-2349. [18] GUO C L, LI C Y, GUO J C, et al. Zero-Reference Deep Curve Estimation for Low-Light Image Enhancement// Proc of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2020: 1777-1786. [19] OKTAY O, SCHLEMPER J, FOLGOC L L, et al. Attention U-Net: Learning Where to Look for the Pancreas[C/OL].[2022-04-25]. https://arxiv.org/pdf/1804.03999v3.pdf. [20] HOWARD A G, ZHU M L, CHEN B, et al. MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications[C/OL].[2022-04-25]. https://arxiv.org/pdf/1704.04861.pdf. [21] BUCHSBAUM G.A Spatial Processor Model for Object Color Perception. Journal of the Franklin Institute, 1980, 310(1): 1-26. [22] DENG J K, GUO J, ZHOU Y X, et al. Retinaface: Single-Stage Dense Face Localisation in the Wild[C/OL].[2022-04-25]. https://arxiv.org/pdf/1905.00641.pdf. [23] LI X, WANG W H, HU X L, et al. Selective Kernel Networks// Proc of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2019: 510-519. [24] RONNEBERGER O, FISCHER P, BROX T.U-Net: Convolutional Networks for Biomedical Image Segmentation// Proc of the International Conference on Medical Image Computing and Computer-Assisted Intervention. Berlin, Germany: Springer, 2015: 234-241. [25] LEE C, LEE C, KIM C S.Contrast Enhancement Based on Layered Difference Representation of 2D Histograms. IEEE Transactions on Image Processing, 2013, 22(12): 5372-5384. [26] WANG S H, ZHENG J, HU H M, et al. Naturalness Preserved Enhancement Algorithm for Non-uniform Illumination Images. IEEE Transactions on Image Processing, 2013, 22(9): 3538-3548. [27] SONG F Y, TAN X Y, LIU X, et al. Eyes Closeness Detection from Still Images with Multi-scale Histograms of Principal Oriented Gradients. Pattern Recognition, 2014, 47(9): 2825-2838. [28] BELHUMEUR P N, JACOBS D W, KRIEGMAN D J, et al. Localizing Parts of Faces Using a Consensus of Exemplars// Proc of the IEEE International Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2011: 545-552. [29] WANG Z, BOVIK A C, SHEIKH H R, et al. Image Quality Assessment: From Error Visibility to Structural Similarity. IEEE Transactions on Image Processing, 2004, 13(4): 600-612. [30] VENKATANATH N, PRANEETH D, BH M C, et al. Blind Image Quality Evaluation Using Perception Based Features// Proc of the 21st National Conference on Communications. Washington, USA: IEEE, 2015. DOI: 10.1109/NCC.2015.7084843. [31] MITTAL A, SOUNDARARAJAN R, BOVIK A C.Making a "Com-pletely Blind" Image Quality Analyzer. IEEE Signal Processing Letters, 2013, 20(3): 209-212.