基于YOLO v5的变电站作业人员着装规范性识别
Identification of dress code of workers in substation based on YOLO v5
-
摘要: 为了对变电站现场作业人员工作服着装规范性进行有效识别,防止作业人员穿短袖或将长袖的袖子卷起,本文采用了YOLO v5实时目标检测算法,针对长袖/短袖两类检测问题,修改分类器,将输出层修改为21维张量。对变电站实际工作现场采集到的7135张样本进行标注,对数据集进行图像增广和色彩调整,并对初始化锚框进行聚类分析。基于YOLO v5在COCO上的预训练模型,利用难例挖掘等手段调整训练目标,最终得到最优的监控场景下工作人员着装规范性的检测模型。实验结果表明,在1000张图片测试集上查准率为97.01%,查全率为96.60%,mAP值为0.8767。在实验环境下平均响应速度达到了38ms。最后,将模型封装为微服务,部署于变电站辅助系统。该着装规范性识别模型速度快、误报少,满足了安全监察需求。Abstract: In order to effectively identify the dress code of the field workers in substation and prevent the workers from wearing short sleeves or rolling up long sleeves, the Yolo v5 real-time object detection algorithm is used in this paper, in order to solve the problem of long-sleeve/short-sleeve detection, the classifier is modified and the output layer is modified to 21-dimensional tensor. The 7135 samples collected from the actual working site of substation are annotated, the data set is enlarged by image and adjusted by color, and the initial anchor frame is cluster analyzed. Based on the Yolo v5 pre-training model on COCO, the training targets are adjusted by means such as hard-negative-mining, and finally a standardized detection model for staff dressing in the optimal monitoring scene is obtained. The experimental results show that on the test set of 1000 images, the precision is 97.01%, the recall is 96.60%, and the mAP value is 0.8767. In the experimental environment, the average response speed reached 38 ms. Finally, the model is encapsulated as a micro-service and deployed in a substation auxiliary system. The identification model of dress code has the advantages of high speed and less false alarm, which meets the requirement of safety inspection.