基于改进YOLOv5的绝缘子快速检测 |
| |
作者姓名: | 黄施懿 董效杰 杨龙欢 王一帆 |
| |
作者单位: | 湖北民族大学智能科学与工程学院 |
| |
基金项目: | 湖北省教育厅科学技术研究计划项目(B2016092); |
| |
摘 要: | 为了解决复杂背景下,绝缘子准确快速识别的实时性问题,提出了一种基于YOLOv5改进的轻量型绝缘子检测算法模型。在网络结构中融入了Shufflenet v2网络和深度卷积模块,通过控制通道数和减少网络层数来减少参数量,采用K-means算法调整anchor框,并提出了改进损失函数DCIoU加速了损失函数的收敛。实验结果表明,改进的YOLOv5算法在参数量上仅有原网络的10%,准确率提高了0.2%,推理速度提升了2帧。
|
关 键 词: | 深度学习 目标检测 绝缘子 YOLOv5 |
|
|