ABSTRACT
The COVID-19 epidemic is still rampant around the world. Wearing masks can effectively block the spread of novel coronavirus, while mask wearing detection can timely remind people in public places to wear masks. In order to solve the problem of low accuracy of mask wearing detection under under dim lighting conditions, an improved YOLOv5 algorithm is proposed. Firstly, the Low Light Intensity Image Enhancement module (LLIIE) was embedded into the original YOLOv5 algorithm to improve the algorithm's night vision ability; Secondly, we added the Convolutional Block Attention Module (CBAM) to enhance feature extraction ability; The CIoU loss function is used to replace the original loss function, which makes the detected target position more accurate. The improved YOLOv5 algorithm is trained and tested on the self-built dataset, and its mean Average Precision, Accuracy and Recall reach 87.1%, 89.3% and 82.7% respectively. The experimental results show that the improved YOLOv5 algorithm has the best detection performance and higher practical value compared with other excellent algorithms in dim light condition.
- Y. Pathak, P. K. Shukla and K. V. Arya, "Deep Bidirectional Classification Model for COVID-19 Disease Infected Patients," in IEEE/ACM Transactions on Computational Biology and Bioinformatics, vol. 18, no. 4, pp. 1234-1241, 1 July-Aug. 2021, doi: 10.1109/TCBB.2020.3009859. ZHANG S F. 2017. Study for Low-Altitude Slow and Small Target Radar Detecting. J. Modern Navigation 8, 6 (2017), 436-440.Google ScholarDigital Library
- Sommana , "Development of a face mask detection pipeline for mask-wearing monitoring in the era of the COVID-19 pandemic: A modular approach," 2022 19th International Joint Conference on Computer Science and Software Engineering (JCSSE), Bangkok, Thailand, 2022, pp. 1-6, doi: 10.1109/JCSSE54890.2022.9836283.Google ScholarCross Ref
- Y. Han, C. Dai and D. Liu, "Detection of Face Mask Wearing for COVID-19 Protection based on Transfer Learning and Classic CNN Model," 2022 IEEE International Conference on Networking, Sensing and Control (ICNSC), Shanghai, China, 2022, pp. 1-7, doi: 10.1109/ICNSC55942.2022.10004063.Google ScholarCross Ref
- T. Hayashi and Y. Kadosaki, "Fast and Accurate Template Matching with Silhouette Masking," 2010 IEEE/ACIS 9th International Conference on Computer and Information Science, Yamagata, Japan, 2010, pp. 258-263, doi: 10.1109/ICIS.2010.115.Google ScholarDigital Library
- A. Krizhevsky, I. Sutskever, and G. E. Hinton. 2012. ImageNet classification with deep convolutional neural networks. J. ACM 60, 5 (May 2012), 84–90. https://doi.org/10.1145/3065386.Google ScholarDigital Library
- Simonyan, Karen and Andrew Zisserman. “Very Deep Convolutional Networks for Large-Scale Image Recognition.” CoRR abs/1409.1556 (2014): n. pag.Google Scholar
- C. Szegedy , "Going deeper with convolutions," 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Boston, MA, USA, 2015, pp. 1-9, doi: 10.1109/CVPR.2015.7298594.Google ScholarCross Ref
- K. He, X. Zhang, S. Ren and J. Sun, "Deep Residual Learning for Image Recognition," 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, 2016, pp. 770-778, doi: 10.1109/CVPR.2016.90.Google ScholarCross Ref
- R. Girshick, J. Donahue, T. Darrell, and J. Malik. 2014. Rich feature hierarchies for accurate object detection and semantic segmentation. IEEE Conf. Compute. Vis. Pattern Recognition, 580-587Google Scholar
- S. Ren, K. He, R. Girshick, and J. Sun. 2015. Faster R-CNN: Towards real-time object detection with region proposal networks. Adv. Neural Inf. Process. Syst, 2015, 91–99.Google Scholar
- W. Liu, D. Anguelov, D. Erhan, C. Szegedy, S. Reed, C.-Y. Fu, and A. C. Berg. 2016. SSD: Single shot multibox detector. Eur. Conf. Comput. Vis. Cham, Switzerland: Springer, 21–37.Google Scholar
- J. Redmon, S. Divvala, R. Girshick, and A. Farhadi. 2016. You only look once: Unified, real-time object detection. IEEE Conf. Comput. Vis. Pattern Recognit. (CVPR), 779–788.Google Scholar
- J. Redmon and A. Farhadi. 2017. YOLO9000: Better, faster, stronger. IEEE Conf. Comput. Vis. Pattern Recognit. (CVPR), 6517–6525.Google Scholar
- J. Redmon and A. Farhadi. 2018. YOLOv3: An incremental improvement. arXiv:1804.02767.Google Scholar
- WEI M J, ZHOU T Y, JI Z L, Detection method of mask wearing in public places based on YOLOv3[J]. Journal of Guangxi Normal University (Natural Science Edition), 2023, 41(1)Google Scholar
- NIU J L, GAO L P, ZHANG L, XIE X Y, 2022. Research on Face Recognition Method of Wearing Mask Based on SSD algorithm. Computer simulation, 1006-9348(2022) 08-0204-04.Google Scholar
- Al-Ameen, Zohair. “Nighttime image enhancement using a new illumination boost algorithm.” IET Image Process. 13 (2019): 1314-1320.Google ScholarCross Ref
- S. Woo, J. Park, J.-Y. Lee, and I. S. Kweon. 2018. CBAM: Convolutional Block Attention Module. in Computer Vision – ECCV 2018, 3–19. https://doi.org/10.1007/978-3-030-01234-21Google ScholarCross Ref
- D. Chen and D. Miao. 2021. Control Distance IoU and Control Distance IoU Loss Function for Better Bounding Box Regression. arXiv, http://arxiv.org/abs/2103.11696Google Scholar
- Liu, S.; Qi, L.; Qin, H.; Shi, J.; Jia, J. Path Aggregation Network for Instance Segmentation. 2018 IEEE/CVF Conference on Computer 830 Vision and Pattern Recognition 2018, pp. 8759–8768.Google Scholar
- H. Rezatofighi, N. Tsoi, J. Gwak, A. Sadeghian, I. Reid and S. Savarese, "Generalized Intersection Over Union: A Metric and a Loss for Bounding Box Regression," 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 2019, pp. 658-666, doi: 10.1109/CVPR.2019.00075.Google ScholarCross Ref
Index Terms
- Mask Wearing Detection in Dim Lighting Conditions based on Improved YOLOv5
Recommendations
The drone detection based on improved YOLOv5
AICCC '22: Proceedings of the 2022 5th Artificial Intelligence and Cloud Computing ConferenceThe wide application of drones not only brings convenience to production and life, but also poses a threat to public safety. Therefore, the detection of s is crucial. However, tiny drones make it difficult to cope with traditional detection methods such ...
Research on Target Detection Method Based on Improved YOLOv5
Web Information Systems and ApplicationsAbstractAt present, various target detection algorithms are used in detection and classification. It is a problem to improve the accuracy and speed of target detection by using deep learning and neural network model to train the parameters of the data ...
Anomaly Detection Model for Key Places Based on Improved YOLOv5
Artificial Intelligence and SecurityAbstractIn recent years, key places such as underground stations and train stations, which are crowded and highly mobile, have become key targets for abnormal behaviour such as violence by some extremists or violent elements. The public safety risks in ...
Comments