A STDAB-DETR Model for Secure Detection of Electrical Construction Sites

Jing Ma,Laijun Chen,Bowen Ren,Suxun Zhu
DOI: https://doi.org/10.1109/icpes59999.2023.10400109
2023-01-01
Abstract:To address the issue of low accuracy and missed detections of safety protection equipment at electrical construction sites, this paper introduces the STDAB-DETR (Swin Transformer Backbone in Dynamic Anchor Boxes for Detection Transformer), a novel safety detection method specifically designed for electrical construction sites. The proposed approach tackles the limitations of traditional convolutional neural networks in purifying important features and extracting information. To overcome these limitations, the Swin Transformer is employed as the backbone, leveraging its hierarchical structure and window-based attention mechanism to effectively capture global contextual information. Additionally, a multi-scale hierarchical structure and a spatial channel attention model are incorporated for efficient feature fusion. To enhance the modeling of irregular-shaped objects, the decoder of the Transformer Integrates Deformable Convolution Networks, enabling the adaptive sampling of sparse high-quality points in the feature space. Furthermore, a novel bounding box loss function combining Smooth L1 and the Complete Intersection over Union Loss (CIOU) is constructed by taking into account the overlap rate, scale, and aspect ratio between the ground truth boxes and the predicted boxes. The case study demonstrates the effectiveness of the proposed model, achieving mAP of 58.6%, A$P_{50}$ of 90.0%, and A$P_{75}$ of 65.2% in complex electrical construction sites environments. These results exhibit improvements of 9.3%, 5.2%, and 13.1%, respectively, compared to the DAB-DETR, which contribute to the reduction of construction-related accidents and the enhancement of worker safety.
What problem does this paper attempt to address?