第21卷第1期2023年3月南京工程学院学报(自然科学版)JournalofNanjingInstituteofTechnology(NaturalScienceEdition)Vol.21ꎬNo.1Mar.ꎬ2023doi:10.13960/j.issn.1672-2558.2023.01.006投稿网址:http://xb.njit.edu.cn基于改进YOLOv5的电网工人防具检测戴云周1ꎬ路红1ꎬ纪陈阳2(1.南京工程学院自动化学院ꎬ江苏南京211167ꎻ2.凌志软件如皋有限公司ꎬ江苏如皋226599)摘要:为了防止电网工人在高空作业时防具穿戴不当而导致安全事故发生ꎬ基于PyTorch框架提出一种改进型的轻量级防具检测算法.首先将YOLOv5s的Backbone用轻量级网络GhostNet来替换ꎬ降低网络复杂程度ꎬ减少参数量ꎻ然后针对工人在高空作业中的目标遮挡问题ꎬ通过改进随机擦除与Mosaic数据增强的方法模拟高空中防具被遮挡的情况ꎻ最后优化loss函数ꎬ采用DistanceIoU替换常用的IoUꎬ解决网络检测框位置偏差大的问题.采用平均精度均值mAP与模型大小来评估本文改进算法与其他算法对防具检测的效果ꎬ试验结果表明了本文改进方法的有效性ꎬ满足在移动端部署的需求.关键词:轻量化ꎻ防具检测ꎻYOLOv5网络中图分类号:TB391.41收稿日期:2022-05-12ꎻ修回日期:2022-05-25基金项目:南京工程学院研究生大学生科技创新项目(TB202217006)作者简介:戴云周ꎬ硕士研究生ꎬ研究方向为计算机视觉.E ̄mail:826265628@qq.com引文格式:戴云周ꎬ路红ꎬ陆慧雯.基于改进YOLOv5的电网工人防具检测[J].南京工程学院学报(自然科学版)ꎬ2023ꎬ21(1):33-38.GridWorkerArmorDetectionBasedonImprovedYOLOv5DAIYunzhouLUHongLUHuiwen1.SchoolofAutomationNanjingInstituteofTechnologyNanjing211167China2.LinkageRugaoSoftwareCo.Ltd.Rugao226599ChinaAbstractInordertopreventsafetyaccidentscausedbyimproperwearingofprotectiveequi...