一种基于改进YOLOv5模型的输电线路金具检测方法
摘要:
本申请公开了一种基于改进YOLOv5模型的输电线路金具检测方法,包括步骤:构建输电线路金具数据集,采集不同类别的金具图像用于模型训练和验证;利用轻量化检测模型YOLOv5作为基础框架,用于检测模型的构建和改进;对YOLOv5模型的主干进行改进,利用可变形卷积改进原始卷积,提高特征提取能力;引入坐标注意力机制模块,增强提取到的特征以用于后续双向特征融合,提高检测精确度;在构建的金具数据集上训练模型并验证,最后通过推理获得预测结果。本发明将改进的轻量检测模型应用于输电线路金具等关键部件的检测,提高了复杂环境下多种金具的检测效果,有效提高了检测的精度。
0/0