基于改进YOLOv5的X射线图像铸件缺陷实时检测

作者：胡哲; 徐承志^*; 雷光波; 徐丽

来源：激光杂志, 2022, 43(05): 54-59.

DOI：10.14016/j.cnki.jgzz.2022.05.054

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

铸件缺陷检测是一项重要的质量管理程序。为了避免人为因素的影响，提高检测精度，对YOLOv5s6的目标检测算法进行改进，用于X射线图像的铸件缺陷检测。首先设计了一种C3CA模块用于捕获跨通道、方向感知和位置感知的信息。然后通过在骨干网络中融合多头自注意力机制来捕获局部与全局信息。最后采用Focal-EIoU Loss损失函数。实验结果表明:在相同训练条件下，改进后YOLOv5s6的AP50值达到了90.2%,F1因子达到了87.8%，相较原始模型分别提高了3.4%和2.3%，具有检测准确率高、实时性强等特点。

• 单位
  湖北工业大学