Optics and Precision Engineering, Volume. 30, Issue 1, 117(2022)

Transfer learning techniques for semantic segmentation of machine vision inspection and identification based on label-reserved Softmax algorithms

Guixiong LIU* and Jian HUANG
Author Affiliations
  • School of Mechanical and Automotive Engineering, South China University of Technology, Guangzhou510640, China
  • show less

    A convolutional neural network (CNN) model for machine vision inspection and identification can identify and measure the components, size, and other features of an object under test. Herein, a fine-tuning transfer learning technique for semantic segmentation based on a label-reserved softmax algorithm was proposed. First, the transfer learning modeling of semantic segmentation for machine vision inspection and identification was performed. Transferring more CNN model weights would reduce the initial loss of the model. Second, a fine-tuning transfer learning method based on label-reserved softmax algorithms was proposed, which could realize fine-tuning transfer learning with all model weights of slightly different detected objects. Experiments based on custom-developed datasets show that the training time for training models to satisfy the requirements of machine vision inspection and identification is reduced from 42.8 min to 30.1 min. Application experiments show that this transfer learning technique enables semi-supervised learning for the inspection of standard component installation, the inspection of missed and mis-installation cases, and the identification of assembly quality. The training time for the transfer learning of new chassis is less than 20.2 min, and the inspection accuracy reaches 100%. The fine-tuning transfer learning technique is effective and satisfies the requirements of machine vision inspection and identification.

    Tools

    Get Citation

    Copy Citation Text

    Guixiong LIU, Jian HUANG. Transfer learning techniques for semantic segmentation of machine vision inspection and identification based on label-reserved Softmax algorithms[J]. Optics and Precision Engineering, 2022, 30(1): 117

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: May. 16, 2021

    Accepted: --

    Published Online: Jan. 20, 2022

    The Author Email: LIU Guixiong (megxliu@scut.edu.cn)

    DOI:10.37188/OPE.20223001.0117

    Topics