多特征融合的YOLOv4-tiny带钢表面缺陷检测方法研究

STUDY ON SURFACE DEFECT DETECTION METHOD OF YOLOV4-TINY STRIP BY MULTI-FEATURE FUSION

  • 摘要: 微小表面缺陷自动识别是带钢生产过程中的研究难点之一。为了提高带钢表面缺陷检测的准确性,提出一种多特征融合的YOLOv4-tiny深度学习方法。引入Inception结构与多尺度信息。提取原始图片的方向梯度直方图特征(HOG),并与主干网络所提取的高层特征相融合,作为特征金字塔结构的输入。实验结果表明,该算法在测试集中带钢表面缺陷mAP达到93.99%,相比原网络提高了13.57百分点,网络参数量相比于原网络减少约21万,网络检测精度有较大的提升。

     

    Abstract: Automatic identification of small surface defects is one of the difficulties in strip production. In order to improve the accuracy of surface defect detection of strip steel, a multi-feature fusion YOLOv4-tiny deep learning method is proposed. The Inception structure and multi-scale information were introduced. The orientation gradient histogram feature (HOG) of the original image was extracted and fused with the high-level features extracted from the backbone network as the input of the feature pyramid structure. The experimental results show that the mAP of surface defects of strip steel in the test concentration is 93.99%, which is 13.57 percentage points higher than that of the YOLOv4-tiny network. The number of network parameters was reduced by about 210 000 compared with that of the YOLOv4-tiny network, and the network detection accuracy is greatly improved.

     

/

返回文章
返回