高级检索

一种轻量级高效X射线焊缝图像缺陷检测方法

A lightweight and efficient X-ray weld image defect detection method

  • 摘要: 针对当前深度学习模型在焊缝缺陷检测工作中成本高、速度慢、不易在终端部署应用等问题,提出一种效果良好的轻量级高效模型(lightweight and high-precision optimized detection model,LHODM). 首先,搭建新型主干网络并设计超轻量卷积ULConv作为网络基础卷积,降低模型深度,用Depthwise操作生成更多有效冗余特征映射,减少参数与运算量. 其次,设计高效轻量级专注模块ELCC,在响应模型轻量化前提下,考虑焊缝缺陷分布特性与成像规律,从水平和垂直两个方向捕获孤立区域关系,结合轻量级上采样算子CARAFE,使模型特征重组时具有更大的感受野,更有效地利用环境周围信息,弥补轻量化造成的精度损失. 最后,为提高收敛速度和损失函数效率,设计OS-IoU损失函数,考虑期望回归向量之间的夹角,重新定义惩罚项及相关性,强化距离损失和形状损失关注程度. 结果表明,LHODM模型检测准确率和检测速度达到91.62%和63.47帧/s,参数量仅为3.99 M,有效解决了焊缝缺陷检测工作中成本高和速度慢的问题.

     

    Abstract: In response to the high cost, slow speed, and difficulty in deploying deep learning models in weld defect detection, this paper proposes a lightweight and high precision optimized detection model (LHODM) with good performance. Firstly, a new backbone network is built and an ultra lightweight convolutional ULConv is designed as the network foundation convolution to reduce model depth, and Depthwise operation was used to generate more effective redundant feature mappings, reducing the amount of parameters and computation. Secondly, an efficient and lightweight dedicated module ELCC is designed. Under the premise of lightweight response model, the distribution characteristics of weld defects and imaging rules are considered, and the relationship between isolated areas is captured from both horizontal and vertical directions.Combined with the lightweight up-sampling operator CARAFE, the model has a larger receptive field during feature recombination, makes more effective use of ambient information, and makes up for the loss of lightweight accuracy. Finally, in order to improve convergence speed and loss function efficiency, an OS-IoU loss function is designed, taking into account the angle between the expected regression vectors, redefining the penalty term and correlation, and strengthening the attention to distance loss and shape loss. The results showed that the LHODM model achieved detection accuracy and speed of 91.62% and 63.47 frames per second, with a model memory cost of only 3.99 million, effectively solving the problems of high cost and slow speed in weld defect detection work.

     

/

返回文章
返回