Laser & Optoelectronics Progress, Volume. 60, Issue 15, 1524001(2023)

Metal Workpiece Surface Defect Segmentation Method Based on Improved U-Net

Yi Wang, Xiaojie Gong*, and Jia Cheng
Author Affiliations
  • College of Electrical Engineering, North China University of Science and Technology, Tangshan 063210, Hebei, China
  • show less

    To solve the problem of low segmentation accuracy of metal workpiece surface defects, we propose a workpiece surface defect segmentation model based on a U-net network combined with a multi-scale adaptive-pattern feature extraction and bottleneck attention module. First, we embed a multi-feature attention aggregation module in the network to improve the utilization of information and extract more relevant features, so as to extract defect targets with high accuracy. Then, the bottleneck attention modules are introduced into the network to increase the weight of defect targets, optimize the extraction of features, and obtain more feature information, thus obtaining better segmentation accuracy. The improved network mean pixel accuracy reaches 0.8749, which is 2.92% higher than the original network. The mean intersection over union reaches 0.8625, an increase of 3.72%. Compared to the original network, the improved network has better segmentation accuracy and segmentation results.

    Tools

    Get Citation

    Copy Citation Text

    Yi Wang, Xiaojie Gong, Jia Cheng. Metal Workpiece Surface Defect Segmentation Method Based on Improved U-Net[J]. Laser & Optoelectronics Progress, 2023, 60(15): 1524001

    Download Citation

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

    Category: Optics at Surfaces

    Received: Jun. 2, 2022

    Accepted: Jul. 26, 2022

    Published Online: Aug. 11, 2023

    The Author Email: Gong Xiaojie (1692994031@qq.com)

    DOI:10.3788/LOP221756

    Topics