高级检索

基于YOLOv10的钛合金表面缺陷检测轻量化算法研究

Lightweight algorithm for titanium alloy surface defect detection based on YOLOv10

  • 摘要: 钛合金表面缺陷检测是保证钛合金产品生产和使用安全的一项重要任务,但是现有检测手段检测效率低,难以部署在终端设备。为了克服现有技术的不足,采用了一种基于YOLOv10n轻量化改进的Titanium-YOLO钛合金表面缺陷检测算法。在主干网络部分采用DualConv替代部分卷积,将C2f-DualConv模块替代传统C2f模块,减小了模型参数量和计算量;采用SlimNeck结构轻量化改进颈网络,提升网络性能的同时降低了模型计算量;采用轻量化检测头litehead替代传统v10Detect,在保证检测精度基本不变的前提下,降低了模型参数量和计算量。结果表明,Titanium-YOLO算法在自主构建钛合金数据集中的平均精度为99.0%,较基准模型提升了2.4%;参数量和计算量分别为2.6×106和7.3×109,较基准模型分别降低了3.7%和13.1%,优于其他实验对比算法。该算法具有高精度和低成本,方便部署在终端设备上,具有较好的工程应用价值。

     

    Abstract:
    Titanium alloy additive manufacturing is one of the disruptive technologies in the manufacturing industry, and titanium alloy products serve as critical materials for national defense equipment such as aircraft and missiles. During production and service, titanium alloy products are susceptible to surface defects such as cracks and scratches caused by harsh external environments. If such defects are not detected and addressed promptly, they may lead to serious consequences. Therefore, surface defect detection for titanium alloys is an important task. However, current detection methods suffer from low efficiency and several other limitations, including difficulties in deployment on terminal devices. To overcome these shortcomings, this paper proposes Titanium-YOLO, an improved surface defect detection algorithm for titanium alloys based on the YOLOv10n model.
    To improve the YOLOv10n base model, key literature was reviewed to identify its fundamental network structure, which comprises the backbone, neck, and detection head. It was determined that developing an efficient detection algorithm relied on multifaceted lightweight improvements to this base structure. Guided by convolutional neural network principles, DualConv was introduced to replace certain standard convolutions in the backbone, and the traditional C2f module was replaced by the C2f-DualConv module. In the neck, the SlimNeck structure was adopted for lightweight enhancement, incorporating key lightweight modules such as GSConv and VoVGSCSP. For the detection head, the lightweight litehead replaced the traditional v10Detect, where DWConv and SPConv substituted standard convolutions to reduce module redundancy. Finally, using a self-constructed titanium alloy dataset and existing laboratory experimental conditions, ablation experiments were conducted to evaluate the performance of each improved module, alongside comparative experiments evaluating the Titanium-YOLO algorithm with mainstream YOLO algorithms.
    Results of the ablation experiments (Table 4, Fig.13) and comparative experiments (Table 5) demonstrated that the improved Titanium-YOLO model had 2.6×106 parameters, a reduction of 3.7% compared to the baseline model, and a computational cost of 7.3×109, which was 13.1% lower. Meanwhile, the improved algorithm achieved a mean average precision (mAP) of 99.0% on the self-constructed titanium alloy dataset, surpassing the baseline by 2.4 percentage points. Furthermore, compared with mainstream YOLO algorithms such as YOLOv5, YOLOv7, and YOLOv8, the improved Titanium-YOLO algorithm demonstrated higher accuracy and a more lightweight design.
    Timely detection of surface defects in titanium alloys is crucial to ensure component safety and prevent serious incidents. Nonetheless, current detection technologies suffer from low efficiency and challenges in deployment on terminal devices. This paper proposes a lightweight Titanium-YOLO detection algorithm, improved from YOLOv10n through the comprehensive integration of DualConv, the SlimNeck structure, and the litehead module. Experimental results show that the Titanium-YOLO algorithm achieves a mAP of 99.0% on the self-constructed titanium alloy dataset, with the number of parameters and computational cost reduced to 2.6×106 and 7.3×109, respectively. The improved algorithm significantly outperforms the baseline and other mainstream models, offering higher accuracy, smaller model sizes, lower computational cost, and simpler deployment. Future work will focus on continued iteration and optimization of the algorithm, with practical implementation and deployment of detection equipment becoming a key research focus.

     

/

返回文章
返回