ISSN 1004-4140
CN 11-3017/P

基于YOLOv8改进算法的铝、镁合金铸件DR图像缺陷检测

Defect Detection in the DR Images of Aluminum and Magnesium Alloy Castings Based on the Improved YOLOv8 Algorithm

  • 摘要: 针对铝、镁合金铸件DR图像背景结构复杂、且存在噪声,导致现有算法对小目标缺陷容易出现漏检的问题,本文提出基于YOLOv8改进算法的铝、镁合金铸件DR图像缺陷检测方法。首先,提出一种融合多尺度增强和对比度受限自适应直方图均衡化的方法,有效解决铝、镁合金铸件DR图像的噪声、低亮度和信息不足的问题。其次,改进YOLOv8网络结构,引入上下文锚点(CAA)注意力机制,更加关注重点缺陷征;引入跨尺度特征融合模块(CCFM),增强多尺度特征的表达能力;改进检测头,提升小目标的检测能力。通过实验结果表明,改进YOLOv8算法在缺陷数据集上的精确率、召回率和均值平均精度(mAP)分别达到92.7%、98.5%和92.4%,检测速度为138 FPS,满足智能化生产线对铝、镁合金铸件缺陷检测的实时性需求。

     

    Abstract: Existing algorithms can miss detecting small target defects owing to the complex background structure and noise in the DR images of aluminum and magnesium alloy castings. Therefore, this study proposes a defect detection method for the DR images of aluminum and magnesium alloy castings based on the improved YOLOv8 algorithm. First, a method integrating multi-scale enhancement and contrast-limited adaptive histogram equalization is proposed to effectively solve the problems of noise, low brightness,and insufficient information in the DR images of aluminum and magnesium alloy castings. Second, the YOLOv8 network structure is improved, and the context anchor attention (CAA) mechanism is introduced to pay more attention to key defect features; the cross-scale feature fusion module (CCFM) is introduced to enhance the expression ability of multi-scale features; the detection head is improved to enhance the detection ability of small targets. Experimental results showed that the precision, recall, and mean average precision (mAP) of the improved YOLOv8 algorithm on the defect dataset reached 92.7%, 98.5%, and 92.4%, respectively, and the detection speed was 138 FPS that met the real-time requirements of intelligent production lines for defect detection in the DR images of aluminum and magnesium alloy castings.

     

/

返回文章
返回