Abstract
In some factories or construction sites, accidents occur because workers do not wearing safety helmets correctly. In order to reduce the accident rate, an improved detection method of safety helmet wearing based on CenterNet algorithm is proposed. The original IOU method is optimized by combining with GIoU, and debug the training model Res/DLA framework in the training process. At the same time, various parameters are adjusted by experiments. In the safety helmet wearing test task, theoretical analysis and experimental results show that mAP (Mean Average Precision) is up to 42.6%, detection rate is increased to 30.3%. Compared with CenterNet, the detection accuracy and detection rate are slightly improved. The proposed algorithm not only meets the real-time performance of detection task in safety helmet wearing detection but also has higher detection accuracy.
Supported by the National Key R & D Program of China (2018AAA0101703) and the key research and development project of Shandong province (2019GNC106093).
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Tingting Z., Jianwu Z., et al.: A survey of image object detection algorithm based on deep learning. Telecommun. Sci. 36(7), 92–106 (2020)
Girshick, R.: Fast R-CNN. In: IEEE International Conference on Computer Vision, pp. 1440–1448 (2015)
Redmon J, Divvala, S., Girshick, R., et al.: You only look once: unified, real-time object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788 (2015)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: IEEE Conference on Computer Vision and Pattern Recognition, pp. 6517–6525 (2017)
Redmon, J., Farhadi, A.: YOLOv3: an incremental improvement. In: IEEE Conference on Computer Vision and Pattern Recognition, pp. 89–95 (2018)
Zhou, X.Y, Wang, D., Philipp, K.: Objects as Points. In: Conference on Computer Vision and Pattern Recognition (CVPR) (2019)
Rezatofighi, H., Tsoi, N., Gwak, J.Y., et al.: Generalized intersection over union: a metric and a loss for bounding box regression. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), IEEE (2020)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 ICST Institute for Computer Sciences, Social Informatics and Telecommunications Engineering
About this paper
Cite this paper
Wang, B., Zhang, Y., Zhao, Q., Shi, S. (2021). An Improved Detection Method of Safety Helmet Wearing Based on CenterNet. In: Fu, W., Xu, Y., Wang, SH., Zhang, Y. (eds) Multimedia Technology and Enhanced Learning. ICMTEL 2021. Lecture Notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering, vol 387. Springer, Cham. https://doi.org/10.1007/978-3-030-82562-1_20
Download citation
DOI: https://doi.org/10.1007/978-3-030-82562-1_20
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-82561-4
Online ISBN: 978-3-030-82562-1
eBook Packages: Computer ScienceComputer Science (R0)