Research on Detection Algorithm of Safety helmet based on improved YOLOX

Weichen Liao

doi:10.54097/fcis.v3i2.7690

Authors

Weichen Liao

DOI:

https://doi.org/10.54097/fcis.v3i2.7690

Keywords:

YOLOX, Helmet detection, Building construction, Target detection

Abstract

In the industrial production, construction and other fields, safety helmet as an important safety protective equipment, the wearing situation of workers personal safety and property safety is of great significance. Therefore, this paper proposes a safety helmet detection method based on the improved YOLOX algorithm. First, the 5000 pictures of safety helmet wearing at construction sites are labelled. The Squeeze and-Excitation module is introduced in the YOLOX network structure. The original Loss function is replaced with varifocal Loss. After experimental verification, compared with the original YOLOX target detection algorithm, our algorithm improves by 2.13 percentage points, enhances the model's focus on key areas and optimizes the model training effect, while the number of model parameters does not increase significantly. In conclusion, our algorithm has a wide range of application prospects and research value.

Downloads

Download data is not yet available.

References

Wang, Y., Zhang, D., Liu, S., Hou, Q., Zhang, C., Chen, B., ... & Huang, F. (2021). YOLOX: Exceeding YOLO Series in 2021. arXiv preprint arXiv:2107.08430

Jie H, Li S , Gang S , et al. Squeeze-and-Excitation Networks[C]// IEEE. IEEE, 2017.

Zhang H, Wang Y , Dayoub F , et al. VarifocalNet: An IoU-aware Dense Object Detector[J]. 2020.

irshick Ross,Donahue Jeff,Darrell Trevor,Malik Jitendra. RegionBased Convolutional Networks for Accurate Object Detection and Segmentation. [J]. IEEE transactions on pattern analysis and machine intelligence,2016,38(1).

Redmon, Joseph, et al. "You only look once: Unified, real-time object detection." Proceedings of the IEEE conference on computer vision and pattern recognition. 2016.

Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.-Y., & Berg, A. C. (2016). SSD: Single shot multibox detector. In European conference on computer vision (pp. 21-37). Springer.

Law H, Deng J . CornerNet: Detecting Objects as Paired Keypoints[J]. International Journal of Computer Vision, 2020, 128(3):642-656.

Liu S, Qi L , Qin H , et al. Path Aggregation Network for Instance Segmentation[C]// IEEE. IEEE, 2018.

Huang L, Yang Y , Deng Y , et al. DenseBox: Unifying Landmark Localization with End to End Object Detection[J]. Computer Science, 2015.

Everingham M, Winn J . The PASCAL Visual Object Classes Challenge 2010 (VOC2010) Development Kit Contents. Springer-Verlag.