基于轻量化YOLOv4的发电机定子表面缺陷检测算法
张凯
罗欣
孙志刚
肖力
华中科技大学人工智能与自动化学院 武汉 430074
摘要：目前大型发电机定子表面缺陷检测主要以抽转子的人工检测为主,存在检测周期长,准确率差等问题,论文提出一种基于轻量化YOLOv4的发电机定子表面缺陷检测算法,以腔内爬壁机器人为载体进行定子缺陷检测.将改进的Mo?bileNetv3作为算法的主干特征提取网络,通过在特征融合层引入CSP结构,融合卷积层和BN层的方法,使得论文算法模型体积较YOLOv4大幅减小.实验结果表明,论文算法在本文发电机定子表面缺陷数据集上的平均检测精度为98.3％,优于原始YOLOv4,模型体积比YOLOv4缩小了84.5％,检测速度提高了45.4％,表明了该方法在嵌入式平台上进行发电机定子缺陷实时检测的应用前景.
关键词：定子缺陷检测MobileNetV3YOLOv4模型轻量化融合卷积层和BN层
分类号：TP301.6(计算技术、计算机技术)
资助基金：国家自然科学基金(51807074)
论文发表日期：2021-04-20
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：7( 686-691,710 )
英文信息