A deep learning-based algorithm for the detection of personal protective equipment.
Personal protective equipment (PPE) is critical for ensuring the safety of construction workers. However, site surveillance images from construction sites often feature multi-size and multi-scale targets, leading to low detection accuracy for PPE in existing models. To address this issue, this paper...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2025-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://doi.org/10.1371/journal.pone.0322115 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850162841972113408 |
|---|---|
| author | Bo Tong Guan Li Xiangli Bu Yang Wang Xingchen Yu |
| author_facet | Bo Tong Guan Li Xiangli Bu Yang Wang Xingchen Yu |
| author_sort | Bo Tong |
| collection | DOAJ |
| description | Personal protective equipment (PPE) is critical for ensuring the safety of construction workers. However, site surveillance images from construction sites often feature multi-size and multi-scale targets, leading to low detection accuracy for PPE in existing models. To address this issue, this paper proposes an improved model based on YOLOv8n.By enriching feature diversity and enhancing the model's adaptability to geometric transformations, the detection accuracy is improved.A Multi-Scale Group Convolution Module (MSGP) was designed to extract multi-level features using different convolution kernels. A Multi-Scale Feature Diffusion Pyramid Network (MFDPN) was developed, which aggregates multi-scale features through the Multiscale Feature Focus (MFF) module and diffuses them across scales, providing each scale with detailed contextual information. A customized Task Alignment Module was introduced to integrate interactive features, optimizing both classification and localization tasks. The DCNV2(Deformable Convolutional Networks v2) module was incorporated to handle geometric scale transformations by generating spatial offsets and feature masks from interactive features, thereby improving localization accuracy and dynamically selecting weights to enhance classification precision.The improved model incorporates rich multi-level and multi-scale features, allowing it to better adapt to tasks involving geometric transformations and aligning with the image data distribution in construction scenarios. Additionally, structured pruning techniques were applied to the model at varying levels, further reducing computational and parameter loads. Experimental results indicate that at a pruning level of 1.5, mAP@0.5 and mAP@0.5:0.95 improved by 3.9% and 4.6%, respectively, while computational load decreased by 21% and parameter count dropped by 53%. The proposed MFD-YOLO(1.5) model achieves significant progress in detecting personal protective equipment on construction sites, with a substantial reduction in parameter count, making it suitable for deployment on resource-constrained edge devices. |
| format | Article |
| id | doaj-art-4b485ee3298a486ba36ea1772da093a6 |
| institution | OA Journals |
| issn | 1932-6203 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-4b485ee3298a486ba36ea1772da093a62025-08-20T02:22:26ZengPublic Library of Science (PLoS)PLoS ONE1932-62032025-01-01205e032211510.1371/journal.pone.0322115A deep learning-based algorithm for the detection of personal protective equipment.Bo TongGuan LiXiangli BuYang WangXingchen YuPersonal protective equipment (PPE) is critical for ensuring the safety of construction workers. However, site surveillance images from construction sites often feature multi-size and multi-scale targets, leading to low detection accuracy for PPE in existing models. To address this issue, this paper proposes an improved model based on YOLOv8n.By enriching feature diversity and enhancing the model's adaptability to geometric transformations, the detection accuracy is improved.A Multi-Scale Group Convolution Module (MSGP) was designed to extract multi-level features using different convolution kernels. A Multi-Scale Feature Diffusion Pyramid Network (MFDPN) was developed, which aggregates multi-scale features through the Multiscale Feature Focus (MFF) module and diffuses them across scales, providing each scale with detailed contextual information. A customized Task Alignment Module was introduced to integrate interactive features, optimizing both classification and localization tasks. The DCNV2(Deformable Convolutional Networks v2) module was incorporated to handle geometric scale transformations by generating spatial offsets and feature masks from interactive features, thereby improving localization accuracy and dynamically selecting weights to enhance classification precision.The improved model incorporates rich multi-level and multi-scale features, allowing it to better adapt to tasks involving geometric transformations and aligning with the image data distribution in construction scenarios. Additionally, structured pruning techniques were applied to the model at varying levels, further reducing computational and parameter loads. Experimental results indicate that at a pruning level of 1.5, mAP@0.5 and mAP@0.5:0.95 improved by 3.9% and 4.6%, respectively, while computational load decreased by 21% and parameter count dropped by 53%. The proposed MFD-YOLO(1.5) model achieves significant progress in detecting personal protective equipment on construction sites, with a substantial reduction in parameter count, making it suitable for deployment on resource-constrained edge devices.https://doi.org/10.1371/journal.pone.0322115 |
| spellingShingle | Bo Tong Guan Li Xiangli Bu Yang Wang Xingchen Yu A deep learning-based algorithm for the detection of personal protective equipment. PLoS ONE |
| title | A deep learning-based algorithm for the detection of personal protective equipment. |
| title_full | A deep learning-based algorithm for the detection of personal protective equipment. |
| title_fullStr | A deep learning-based algorithm for the detection of personal protective equipment. |
| title_full_unstemmed | A deep learning-based algorithm for the detection of personal protective equipment. |
| title_short | A deep learning-based algorithm for the detection of personal protective equipment. |
| title_sort | deep learning based algorithm for the detection of personal protective equipment |
| url | https://doi.org/10.1371/journal.pone.0322115 |
| work_keys_str_mv | AT botong adeeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT guanli adeeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT xianglibu adeeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT yangwang adeeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT xingchenyu adeeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT botong deeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT guanli deeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT xianglibu deeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT yangwang deeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment AT xingchenyu deeplearningbasedalgorithmforthedetectionofpersonalprotectiveequipment |