基于改进YOLO v5s的轻量化高压开关柜识别模型研究
Modified YOLO v5s-based Lightweight Identification Model for High-voltage Switchgears作者机构:国网河南省电力公司新野县供电公司河南南阳473500
出 版 物:《电工技术》 (Electric Engineering)
年 卷 期:2024年第15期
页 面:27-32页
学科分类:081203[工学-计算机应用技术] 08[工学] 0835[工学-软件工程] 0812[工学-计算机科学与技术(可授工学、理学学位)]
主 题:开关柜检测 YOLO v5 特征提取 小目标检测 损失函数
摘 要:针对开关柜指示灯尺寸小、分布密集造成检测困难等问题,提出一种基于YOLO v5改进模型。使用轻量化主干网络模型MobileNet-V3,降低边缘端部署要求。改进多尺度特征融合方式扩大感受野,提高小尺寸目标检测效果。将损失函数替换为WIoU,改善模型难易样本不平衡问题。增加数据集多样性,采用迁移学习加快模型收敛速度和训练效率。提出的算法满足所需模型小、精度高、检测速度快等要求,为开关柜实时检测提供一种解决方法。