Safety helmet wearing detection based on EfficientDet algorithm
Xinlei Fan, Fan Wang, Shuai Pang, Jiaxing Wang, Wenjing Wang · 2nd International Conference on Artificial Intelligence, Automation, and High-Performance Computing (AIAHPC 2022) · 2022
In safety production work and workplaces, many safety accidents occur because workers do not wear safety helmets. In order to reduce the safety accidents caused by not wearing a helmet, this paper proposes a helmet wearing detection method based on the EfficientDet algorithm. First, the K-Means++ clustering algorithm is used to optimize the initial cluster centers, then, the SeparableConv2D network is introduced, combined with the simple and efficient Bi-directional Feature Pyramid Network (BiFPN) proposed in the EfficientDet algorithm to extract image feature maps, finally, the CC-Loss function is used as the classification loss function to constrain the specific relationship between classes and channels, and maintain the separability within and between classes, thereby improving the accuracy of model detection. The experimental results on the open source helmet data set show that under different experimental conditions, the improved efficientset model has better detection accuracy and speed, and can better meet the safety helmet wearing detection.