基于改进YOLOv7的医用防护用品穿戴检测

WEARING DETECTION OF MEDICAL PROTECTIVE EQUIPMENT BASED ON IMPROVED YOLOV7

  • 摘要: 提出一种基于YOLOv7改进的算法模型以进一步提高当前医用防护用品是否正确穿戴的检测精度。在YOLOv7骨干网络中加入CBAM注意力机制,结合通道和空间注意力机制模块以提升模型对医用防护物品的关注能力。采用EIoU作为损失函数以加速预测框的回归速度,并提高模型鲁棒性。实验结果表明:与YOLOv7相比,在自制的医用防护用品穿戴数据集上该方法精确度和mAP@0.5分别提升了6.8百分点和5.2百分点。因此,该模型可有效地提高医用防护用品正确穿戴与否的检测效率。

     

    Abstract: This study proposes an improved algorithm model based on YOLOv7 to further improve the detection accuracy of wearing medical protective equipment correctly. In this work, the CBAM attention mechanism was added to the YOLOv7 backbone network with the channel and spatial attention mechanism modules combined to improve its ability to pay attention on the medical protective equipment. In addition, EIoU was utilized as the loss function to speed up the regression speed of the prediction box and improve the robustness of the model. The experimental results show that compared with original YOLOv7, the accuracy and mAP@0.5 of this improved method on the self-made medical protective equipment wearing dataset are increased by 6.8 and 5.2 percentage points, respectively. Therefore, the improved algorithm model based on YOLOv7 proposed in this paper can effectively improve the detection efficiency of wearing medical protective equipment correctly.

     

/

返回文章
返回