一种用于无人机巡检绝缘子的轻量型目标检测方法

    公开(公告)号:CN116385911A

    公开(公告)日:2023-07-04

    申请号:CN202310378402.4

    申请日:2023-04-11

    Abstract: 本发明公开了一种用于无人机巡检绝缘子的轻量型目标检测方法,首先提出一种针对航拍绝缘子图像的旋转矩形框标注方法,对绝缘子图像进行标注;然后基于YOLOv5模型提出一种改进的GhostConv模块,通过ghost卷积大幅降低模型的网络参数,通过Hardswish函数增强模型的数值稳定性;提出一种改进的C3Ghost模块,通过轻量化网络设计减少模型的参数量和浮点计算量;在特征提取和特征融合模块之间引入SE注意力机制,并通过分配权重的方式来抑制对当前任务不重要的特征通道的关注程度;通过引入CSL以分类代替回归来避免旋转角度损失计算时由于边界问题带来的损失突增问题;提出一种高精度旋转框回归损失函数R‑EIoU Loss以此来加速本发明模型的训练过程和提高模型的检测精度。

Patent Agency Ranking