基于SGA与SE-F的轻量级输电线路绝缘子缺陷检测算法

    公开(公告)号:CN119228777A

    公开(公告)日:2024-12-31

    申请号:CN202411394401.X

    申请日:2024-10-08

    Inventor: 李莹 朱昌飞

    Abstract: 基于SGA与SE‑F的轻量级输电线路绝缘子缺陷检测算法,首先,在YOLOv5网络基础上引入全新的注意力机制SGA,增强待检测缺陷目标的显著度,以解决微小绝缘子缺陷特征难以提取的问题;其次,提出了同尺度加权融合模块SE‑f,用于替换特征融合网络中的Concat模块,提高了网络对低层位置信息和高层语义信息的特征的融合能力;再次,对改进后YOLOv5通过稀疏化训练和通道剪枝,进一步优化模型结构;同时利用知识蒸馏引导模型进行知识学习,提高检测精度。本发明有益效果为:本发明首先通过引入注意力机制、构建新型特征融合方式对初始模型进行改进,然后结合通道剪枝算法实现模型轻量化,最后构建知识蒸馏模型提高剪枝后模型检测精度。

Patent Agency Ranking