首页 | 本学科首页   官方微博 | 高级检索  
     

基于改进YOLOv5的X射线图像铸件缺陷实时检测
引用本文:胡哲,徐承志,雷光波,徐丽. 基于改进YOLOv5的X射线图像铸件缺陷实时检测[J]. 激光杂志, 2022, 43(5): 54-59. DOI: 10.14016/j.cnki.jgzz.2022.05.054
作者姓名:胡哲  徐承志  雷光波  徐丽
作者单位:湖北工业大学计算机学院,武汉 430068
基金项目:湖北省教育厅科学技术研究计划项目(No.B2019049);
摘    要:铸件缺陷检测是一项重要的质量管理程序。为了避免人为因素的影响,提高检测精度,对YOLOv5s6的目标检测算法进行改进,用于X射线图像的铸件缺陷检测。首先设计了一种C3CA模块用于捕获跨通道、方向感知和位置感知的信息。然后通过在骨干网络中融合多头自注意力机制来捕获局部与全局信息。最后采用Focal-EIoU Loss损失函数。实验结果表明:在相同训练条件下,改进后YOLOv5s6的AP50值达到了90.2%,F1因子达到了87.8%,相较原始模型分别提高了3.4%和2.3%,具有检测准确率高、实时性强等特点。

关 键 词:铸件缺陷检测  多头自注意力机制  YOLOv5s6  X射线图像

Real-time detection of casting defect in X-ray images based on improved YOLOv5
HU Zhe,XU Chengzhi,LEI Guangbo,XU Li. Real-time detection of casting defect in X-ray images based on improved YOLOv5[J]. Laser Journal, 2022, 43(5): 54-59. DOI: 10.14016/j.cnki.jgzz.2022.05.054
Authors:HU Zhe  XU Chengzhi  LEI Guangbo  XU Li
Abstract:
Keywords:
本文献已被 万方数据 等数据库收录!
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号