一种改进YOLOv7-tiny的轻量级绝缘子缺陷检测方法

    公开(公告)号:CN117392102A

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

    申请号:CN202311443526.2

    申请日:2023-11-01

    Abstract: 本发明公开了一种改进YOLOv7‑tiny的轻量级绝缘子缺陷检测方法,包括:获取缺陷绝缘子图像并构建输电线路绝缘子缺陷图像数据集。基于YOLOv7‑tiny构建轻量化绝缘子缺陷检测模型,输入端引入数据增强策略。特征提取网络部分采用Mish激活函数替换ReLU激活函数,使用ShuffleNet‑v2作为特征提取模块获取特征图。颈部网络采用特征聚合网络获取语义信息并进行特征融合,引入深度可分离卷积和Ghost卷积降低模型计算量。检测头嵌入CBAM注意力机制重点关注目标区域的重要特征,同时抑制背景特征。最后将模型在Windows系统上进行训练轻量化绝缘子缺陷检测模型并进行测试。本发明设计了一种改进YOLOv7‑tiny的轻量级绝缘子缺陷检测方法,提供了一种轻便实时的绝缘子缺陷检测方法。

Patent Agency Ranking