Download pdf
Two-stage helmet security detection model combining CNN and MTCNN algorithm
- 1 The University of Hong Kong
* Author to whom correspondence should be addressed.
Abstract
Physical safety is of utmost importance for workers in an industrial and construction environment. Fatalities related to the violation of wearing a safety helmet account for a large proportion of occupational deaths. Hence, the requirement of a high-efficiency automatic system to monitor and promote wearing helmets is of vital importance, reducing hours from manual monitoring, facilitating efficient education and enforcement campaigns that increase industrial safety. In this paper, a combined Convolutional Neural Network (CNN) algorithm and Multi-Task CNN (MTCNN) to monitor violations of safety helmets is proposed. All images are sourced from industrial and construction locations, undergoing various image processing techniques before model detection. The proposed model includes the MTCNN locating individuals' heads from the picture data, followed by a CNN assessment to determine helmet wearing. Under the same evaluation criterion, the result, where an obvious detection accuracy improvement is shown, illustrates the effectiveness of the proposed model in the comparative study. Therefore, the research demonstrates the effectiveness of incorporating multi-scale feature extraction techniques for enhancing model performance. This establishes the groundwork for future research within this domain, facilitating the creation of more advanced helmet detection or other relevant items recognition systems that can be utilized in real-world applications.
Keywords
CNN, MTCNN, Helmet Detection, Image Processing
[1]. Kamboj A Nilesh P 2020 Safety helmet detection in industrial environment using deep learning Proceedings of the 9th International Conference on Information Technology Convergence and Services (ITCSE 2020)
[2]. Maas A I R et al 2022 Traumatic brain injury: progress and challenges in prevention clinical care, and research The Lancet Neurology
[3]. Van Z Adriaan H Van D 1985 Residual complaints of patients two years after severe head injury Journal of Neurology Neurosurgery & Psychiatry 48(1): pp 21-28
[4]. Liu W et al 2016 SSD: single shot multibox detector Computer Vision ECCV 2016 - 14th European Conference, Amsterdam 2016 Proceedings pp 21–37
[5]. Redmon J Divvala S K Girshick R B and Farhadi A 2016 You only look once: Unified real-time object detection 2016 IEEE Conference Computer Vision and Pattern Recognition (CVPR) pp 779–788
[6]. Huang J Fathi A Rathod V Fischer I and Sun C 2017 Speed/accuracy trade-offs for modern convolutional object detectors Computer Vision and Pattern Recognition arXiv:1611.10012
[7]. Howard A Wang W Zhu M Weyand T Chen B and Kalenichenko B April 2017 MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications Computer Vision and Pattern Recognition arXiv:1704.04861
[8]. Dataset https://www.kaggle.com/datasets/andrewmvd/hard-hat-detection
[9]. Shorten C and Khoshgoftaar T M 2019 A survey on Image Data Augmentation for Deep Learning Journal of Big data 6: p 60
[10]. Phung Rhee 2019 A High-Accuracy Model Average Ensemble of Convolutional Neural Networks for Classification of Cloud Image Patches on Small Datasets Applied Sciences 9: p 4500
[11]. Zhang Z et al 2019 Bag of freebies for training object detection neural networks arXiv preprint arXiv:1902.04103
Cite this article
He,J. (2024). Two-stage helmet security detection model combining CNN and MTCNN algorithm. Applied and Computational Engineering,38,259-266.
Data availability
The datasets used and/or analyzed during the current study will be available from the authors upon reasonable request.
Disclaimer/Publisher's Note
The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of EWA Publishing and/or the editor(s). EWA Publishing and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.
About volume
Volume title: Proceedings of the 2023 International Conference on Machine Learning and Automation
© 2024 by the author(s). Licensee EWA Publishing, Oxford, UK. This article is an open access article distributed under the terms and
conditions of the Creative Commons Attribution (CC BY) license. Authors who
publish this series agree to the following terms:
1. Authors retain copyright and grant the series right of first publication with the work simultaneously licensed under a Creative Commons
Attribution License that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this
series.
2. Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the series's published
version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgment of its initial
publication in this series.
3. Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and
during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See
Open access policy for details).