基于YOLOv5s的配电台区施工多目标检测方法OACSTPCD
Multi-object Detection Method of Distribution Platform Construction Based on YOLOv5s
配电台区建设工程繁多,施工人员作业标准化、规范化程度低,利用目标检测算法对施工过程管控,可有效保证工程质量.常用目标检测算法对设备存储与运算能力要求高,因此如何将算法轻量化部署到边端设备成为研究重点.为提升配电台区设备施工识别的检测精度,同时考虑模型轻量化的需求,文章提出基于YOLOv5s改进的配电台区施工多目标检测算法.首先,利用改进的Res2Net网络的细颗粒、多尺度特征提取bottle2neck模块,实现图像特征多尺度提取,保证模型精确度和实现轻量化;其次,在bottle2neck模块基础上,提出检测精度更高的B4-Cat优化模型;最后,使用某地区提供的配电台区建设数据,验证模型的优越性.结果表明,所提方法与现有算法相比,模型参数和计算量降低25%以上,总体平均精度指标超过81%,效果优于常用的深度可分离卷积轻量化方法,有利于提高配电台区施工智能化管控水平.
There are many construction projects in the distribution platform area,and the standardization and standardization of the construction personnel are low. The use of object detection algorithm to control the construction process can effectively ensure the quality of the project. Common object detection algorithms require high storage and computing power of devices,so how to deploy lightweight algorithms to edge devices has become the focus of research. In order to improve the detection accuracy of equipment construction identification in distribution station area and consider the demand of model lightweight,this paper proposes a multi-object detection algorithm based on YOLOv5s. Firstly,the bottle2neck module of improved Res2Net network was used to extract fine particles and multi-scale features to achieve multi-scale image feature extraction,ensuring model accuracy and lightweight. Secondly,based on bottle2neck module,a B4-Cat optimization model with higher detection accuracy is proposed. Finally,the advantages of the model are verified by the data of distribution station construction provided by certain region. The results show that compared with the existing algorithms,the model parameters and calculation amount of the proposed method are reduced by more than 25%,and the mAP index is more than 81%,which is better than the commonly used depth separable convolutional lightweight method,and is conducive to improving the intelligent management and control level of distribution station construction.
张天明;王金丽;李佳;段祥骏;冯德志;杨乐
中国电力科学研究院有限公司,北京市海淀区 100192
电子信息工程
配电台区YOLOv5Res2Net多尺度特征提取轻量化目标检测
power distribution areaYOLOv5Res2Netmulti-scale feature extractionlight weightobject detection
《电力信息与通信技术》 2024 (007)
59-67 / 9
国家电网有限公司总部科技项目资助"基于BIM的配网工程建设过程管控关键技术研究"(5400-202255155A-1-1-ZN).
评论