YOLOX-αSMV algorithm for surface defect detection of strip steel material
DOI:
CSTR:
Author:
Affiliation:

Clc Number:

Fund Project:

This work is supported by the National Natural Science Foundation of China (61861016), and the Science and Technologies Project of Jiangxi Province (20161BBE50081).

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In order to solve the problems of insufficient feature extraction and weak ability of multi-target defect detection of YOLOX algorithm in steel surface defect detection, a multi-dimensional feature fusion strip material surface defect detection algorithm based on improved loss function is proposed. First of all, apply SPP_SF to the Backbone part to retain multi-scale feature information and improve classification accuracy. Secondly, the multi-dimensional feature fusion module MDFFM is added in the Neck part to integrate the channel, space and position information into the feature vector to strengthen the feature ex-traction ability of the algorithm. Finally, the introduction of Varifocal Loss and α-CIoU weighted positive and negative samples to improve the regression accuracy of the prediction box. The experimental results show that YOLOX-αSMV in NEU-DET data set mAP@0.5:0.95 reaches 47.54%, which is 3.43% higher than YOLOX algorithm. While keeping the detection speed basically unchanged, the ability to identify and locate fuzzy defects and small target defects is significantly improved.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:June 05,2023
  • Revised:August 27,2023
  • Adopted:September 01,2023
  • Online: March 26,2024
  • Published:
Article QR Code