高级检索

一种基于卷积神经网络的轻量级焊缝缺陷识别算法

A Lightweight Weld Defect Identification Algorithm Based on Convolutional Neural Network

  • 摘要: 针对传统焊缝缺陷检测在处理大量焊缝缺陷数据,存在焊缝缺陷识别效率和准确率低等问题,提出1种基于卷积神经网络的轻量级焊缝缺陷识别算法。在原MobileNetV3算法的基础上,引入fire模块对算法进行改进,减小算法的参数量;同时引入ECA(efficient channel attention)模块加强对特征通道的学习,使整个算法合理分配计算资源,整体提高算法的特征提取能力;将改进算法与常见分类模型算法在焊缝缺陷测试数据中进行对比实验,验证改进算法的可行性。结果表明:相比于原算法与其他分类模型算法,改进算法在fire模块和ECA模块的双重作用下,焊缝缺陷识别准确率进一步提升,对凹陷、孔洞、毛刺等焊缝缺陷的平均识别准确率达98.53%;改进的MobileNetv3算法在保证较高识别准确率的情况下,参数量和浮点运算量也显著降低,还有利于模型轻量化部署。

     

    Abstract: In response to the issues of low efficiency and accuracy in traditional weld defect detection when processing large volumes of weld defect data, a lightweight weld defect recognition algorithm based on convolutional neural networks was proposed. Building on the original MobileNetV3 algorithm, the fire module was introduced to reduce the number of parameters in the algorithm. Additionally, the ECA (efficient channel attention) module was incorporated to enhance the learning of feature channels, enabling the algorithm to allocate computational resources more effectively and improve overall feature extraction capabilities. The improved algorithm was compared with common classification model algorithms on weld defect test data to validate its feasibility. The results show that, compared to the original algorithm and other classification model algorithms, the improved algorithm, with the combined effects of the fire module and the ECA module, achieves a further increase in weld defect recognition accuracy, with an average recognition accuracy of 98.53% for defects such as depressions, holes, and burrs. The improved MobileNetV3 algorithm not only maintains high recognition accuracy but also significantly reduces the number of parameters and floating-point operations, facilitating the lightweight deployment of the model.

     

/

返回文章
返回