Multi-level decoding neural network for pitting detection of ball screw
DOI:
CSTR:
Author:
Affiliation:

1.Equipment Reliability Institute, Shenyang University of Chemical Technology,Shenyang 110142, China; 2.School of Mechanical and Power Engineering, Shenyang University of Chemical Technology,Shenyang 110142, China

Clc Number:

TP391.4

Fund Project:

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

    Due to the small pitting area of the ball screw and the serious environmental interference, defects are difficult to detect in time. Therefore, a Multi-level decoding neural network is proposed to realize the segmentation of pitting defects in ball screws. The network consists of an encoder, a multi-level decoder and a Multi-scale Attention module. The encoder is composed of Resnet34, and the Ghost module is introduced to build a lightweight multi-level decoder. In order to fuse multi-scale features and filter redundant information, the Multi-scale Attention module is designed. A hybrid loss function composed of BCE function, IOU and SSIM function is used to train the network. Experiments on the ball screw defect dataset show that Multi-level decoding neural network achieves 0.770 3 in the maxFβ metrics, compared with other methods, which achieves better segmentation results, and the processing time of a single image is 26 ms. It provides a new method for real-time segmentation of ball screw pitting defects.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: April 24,2024
  • Published:
Article QR Code