高级检索
薛均晓, 程君进, 张其斌, 郭毅博, 鲁爱国, 李鉴, 万曦, 徐静. 改进轻量级卷积神经网络的复杂场景口罩佩戴检测方法[J]. 计算机辅助设计与图形学学报, 2021, 33(7): 1045-1054. DOI: 10.3724/SP.J.1089.2021.18635
引用本文: 薛均晓, 程君进, 张其斌, 郭毅博, 鲁爱国, 李鉴, 万曦, 徐静. 改进轻量级卷积神经网络的复杂场景口罩佩戴检测方法[J]. 计算机辅助设计与图形学学报, 2021, 33(7): 1045-1054. DOI: 10.3724/SP.J.1089.2021.18635
Xue Junxiao, Cheng Junjin, Zhang Qibin, Guo Yibo, Lu Aiguo, Li Jian, Wan Xi, Xu Jing. Improved Efficient Convolutional Neural Networks for Complex Scene Mask-Wearing Detection[J]. Journal of Computer-Aided Design & Computer Graphics, 2021, 33(7): 1045-1054. DOI: 10.3724/SP.J.1089.2021.18635
Citation: Xue Junxiao, Cheng Junjin, Zhang Qibin, Guo Yibo, Lu Aiguo, Li Jian, Wan Xi, Xu Jing. Improved Efficient Convolutional Neural Networks for Complex Scene Mask-Wearing Detection[J]. Journal of Computer-Aided Design & Computer Graphics, 2021, 33(7): 1045-1054. DOI: 10.3724/SP.J.1089.2021.18635

改进轻量级卷积神经网络的复杂场景口罩佩戴检测方法

Improved Efficient Convolutional Neural Networks for Complex Scene Mask-Wearing Detection

  • 摘要: 针对复杂光照和人脸倾斜条件下口罩佩戴检测准确率低的问题,提出一种利用轻量级卷积神经网络在复杂环境下的口罩佩戴检测方法.该方法利用难样本挖掘预训练学习更多的人脸特征,结合多任务级联卷积神经网络先判别是否有人脸信息,对其进行精准的人脸定位;在特征金字塔网络中添加注意力机制,增强了人脸关键点信息的权重,利用轻量级神经网络将口罩佩戴检测视为简单的二分类问题进行判断.在TensorFlow的环境下完成了数据训练、数据预处理、与AIZOO方法对比实验,收集建立了包含816张图片的数据集进行标注并训练;在对数据进行预处理操作时先将图片设定为固定大小以降低运算量,提高检测速度,再利用图像增强算法进行扭曲处理提高模型的鲁棒性.在此基础上,利用MTCNN检测图片中的人脸并对其进行修正和归一化操作,然后传入神经网络并利用已经训练好的模型进行检测.实验结果表明,在复杂光照和人脸倾斜等复杂条件下,文中方法的准确率分别达到83%和91%,可以有效地进行口罩佩戴检测.

     

    Abstract: To solve the problem about low accuracy of mask wear detection under complex lighting and face lean conditions,a method of mask wear detection under intricate environment using efficient convolutional neural network is proposed,which uses pre-training such as hard negative mining to learn more samples of face feature,utilize multi-task convolutional neural networks(MTCNN)to estimate the possibility of face information,and get accurate face location.With attention mechanism in feature pyramid network,enhancing the weight of key points on human face,employing efficient neural network detection will be wore on mask-wearing detection as a simple binary classification problem.Under the environment of TensorFlow platform,not only data training,data preprocessing,but also the contrast experiment with AIZOO method are completed.A data set containing with 816 images is collected,marked and trained.During the data preprocessing,images are set as fixed size to reduce the amount of computation and promote the detection speed.Then,image enhancement algorithm is used to conduct distortion processing to improve the robustness of this model.On this basis,MTCNN is used to detect the face information in pictures,modify and normalize all data,then put them into neural network and the trained model to detection.The experimental results show that under complex conditions such as complex lighting and face tilt,the accuracy can reach 83%and 91%respectively,which means can accurately detect whether wearing a mask.

     

/

返回文章
返回