ABSTRACT
Aiming at the problem that the accuracy of mask-wearing detection is poor and it is difficult to deploy in low-cost equipment, this paper has proposed a mask-wearing detection algorithm based on improved YOLOv7. By adding an attention mechanism to the backbone network and introducing the partial convolution method, the algorithm enhances the perception ability of the network to small targets, improves the accuracy of mask-wearing detection, reduces the parameters and calculation of the network, and improves the efficiency of feature extraction of the network. By integrating Distribution Shift Convolution in the head network, the required memory and equipment requirements are reduced while the operation speed is also improved. The experimental results show that, compared to the original model, the mean average precision of the improved YOLOv7 network improved by 0.5%, the parameters are reduced by 20%, the computation only accounts for 51% while the model size is lower to 63%and the reasoning speed remains basically unchanged. These improvements can be applied in the current mask-wearing monitoring task and provide a useful reference for future object detection tasks.
- Song Yao, Liu Jun. Improved U-Net COVID-19 image segmentation method [J]. Computer Engineering and Application, 2021, 57(19):9.Google Scholar
- Wang Xinran, Tian Qichuan, Zhang Dong. A summary of the research on wearing and detecting face masks [J]. Computer Engineering and Application, 2022, 58(10):13-26.Google Scholar
- Viola P . Rapid Object Detection using a Boosted Cascade of Simple Features[J]. Proc. IEEE CVPR 2001, 2001.Google ScholarCross Ref
- Viola P, Jones M J . Robust Real-Time Face Detection[J]. International Journal of Computer Vision, 2004, 57(2):137-154.Google ScholarDigital Library
- Dalal N, Triggs B . Histograms of Oriented Gradients for Human Detection[C]// IEEE Computer Society Conference on Computer Vision & Pattern Recognition. IEEE, 2005.Google Scholar
- Felzenszwalb, Pedro, F, Object Detection with Discriminatively Trained Part-Based Models.[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2010, 32(9):1627-1645.Google ScholarDigital Library
- Bochkovskiy A, Wang C Y, Liao H . YOLOv4: Optimal Speed and Accuracy of Object Detection[J]. 2020.Google Scholar
- Wang C Y, Bochkovskiy A, Liao H . YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[J]. arXiv e-prints, 2022.Google Scholar
- Berg A C , Fu C Y , Szegedy C , SSD: Single Shot MultiBox Detector: 10.1007/978-3-319-46448-0_2[P]. 2015.Google Scholar
- Yu L , Park E , Berg A C , Proceedings - 2015 IEEE International Conference on Computer Vision, ICCV 2015[J]. IEEE, 2016.Google Scholar
- Ren S, He K, Girshick R, Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2017, 39(6):1137-1149.Google ScholarDigital Library
- Zhang Xiubao, Lin Ziyuan, Tian Wanxin, Face mask recognition technology in all-weather natural scenes [J]. China Science: Information Science, 2020, 50(7):11.Google Scholar
- Niu Zuodong, Qin Tao, Li Handong, Improved natural scene mask-wearing detection algorithm of RetinaFace [J]. Computer Engineering and Application, 2020, 56(12):7.Google Scholar
- Chen Zhaojun, Chu Jun, Zeng Lunjie. Multi-category mask-wearing detection based on dynamic weighted category balance loss [J]. Journal of Graphics, 2022, 43(4):9.Google Scholar
- Mao Xiaobo, Xu Xiangyang, Li Nan, The entrance guard system of mask-wearing detection based on improved SSD and Jetson Nano [J]. Journal of Zhengzhou University (Engineering Edition), 2021(006):042.Google Scholar
- CHEN J, KAO S hong, HE H, Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks[Z]. 2023Google Scholar
- Gennari M , Fawcett R , Prisacariu V A . DSConv: Efficient Convolution Operator[J]. 2019.Google Scholar
- Woo S, Park J, Lee J Y, Cbam: Convolutional block attention module[C]//Proceedings of the European conference on computer vision (ECCV). 2018: 3-19.Google Scholar
- Wang Q , Wu B , Zhu P , ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks[C]// 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2020.Google Scholar
- Chen J, Lu Z, Liao Q. XSepConv: extremely separated convolution for efficient deep networks with large kernels[C]//Thirteenth International Conference on Digital Image Processing (ICDIP 2021). SPIE, 2021, 11878: 160-169.Google Scholar
Index Terms
- A mask-wearing detection algorithm based on improved YOLOv7
Recommendations
Helmet detection algorithm based on improved yolov7
CSAI '23: Proceedings of the 2023 7th International Conference on Computer Science and Artificial IntelligenceIn order to solve the problems of low detection accuracy, wrong detection and missed detection in the traditional target detection algorithm in the safety helmet detection task, a safety helmet detection algorithm based on improved YOLOv7 is proposed. ...
Efficient Real-Time Dense Pedestrian Detector Based on Improved YOLOv7-tiny
ICAIP '23: Proceedings of the 2023 7th International Conference on Advances in Image ProcessingThe state-of-the-art YOLO framework strikes an excellent balance between speed and accuracy and has become one of the most effective object detection algorithms. However, when performing dense pedestrian detection tasks, the network is computationally ...
An efficient detection method for litchi fruits in a natural environment based on improved YOLOv7-Litchi
Highlights- An improved YOLOV7-Litchi for litchi detection in complex environments was proposed.
- A dataset of litchi fruits in complex orchard environments was established.
- The proposed method achieves satisfactory results in lightweight, ...
AbstractThe identification and detection of small fruit targets combined with mechanized harvesting can effectively solve the problems of high labor intensity, high labor cost and low harvesting efficiency in modern litchi harvesting. The orchard ...
Comments