关键词:
配电台区
YOLOv5
Res2Net
多尺度特征提取
轻量化
目标检测
摘要:
配电台区建设工程繁多,施工人员作业标准化、规范化程度低,利用目标检测算法对施工过程管控,可有效保证工程质量。常用目标检测算法对设备存储与运算能力要求高,因此如何将算法轻量化部署到边端设备成为研究重点。为提升配电台区设备施工识别的检测精度,同时考虑模型轻量化的需求,文章提出基于YOLOv5s改进的配电台区施工多目标检测算法。首先,利用改进的Res2Net网络的细颗粒、多尺度特征提取bottle2neck模块,实现图像特征多尺度提取,保证模型精确度和实现轻量化;其次,在bottle2neck模块基础上,提出检测精度更高的B4-Cat优化模型;最后,使用某地区提供的配电台区建设数据,验证模型的优越性。结果表明,所提方法与现有算法相比,模型参数和计算量降低25%以上,总体平均精度指标超过81%,效果优于常用的深度可分离卷积轻量化方法,有利于提高配电台区施工智能化管控水平。