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
    References(16)

    [1] Tabernik D, Šela S, Skvarč J et al. Segmentation-based deep-learning approach for surface-defect detection[J]. Journal of Intelligent Manufacturing, 31, 759-776(2020).

    [2] Hu G H, Huang J F, Wang Q H et al. Unsupervised fabric defect detection based on a deep convolutional generative adversarial network[J]. Textile Research Journal, 90, 247-270(2019).

    [3] Liu X B, Zhang Y W. Ore Image Segmentation method of conveyor belt based on U-net and Res_UNet models[J]. Journal of Northeastern University (Natural Science), 40, 1623-1629(2019).

    [4] Liu J W, Liu Q H, Li X O et al. Improved colonic polyp segmentation method based on double U-shaped network[J]. Acta Optica Sinica, 41, 1810001(2021).

    [5] Zhang H L, Li Q, Guan X. An improved three-dimensional dual-path brain tumor image segmentation network[J]. Acta Optica Sinica, 41, 0310002(2021).

    [6] Xiao Z T, Liu B W, Geng L et al. Segmentation of lung nodules using improved 3D-UNet neural network[J]. Symmetry, 12, 1787(2020).

    [7] Xu G L, Mao J. Few-shot segmentation on mobile phone screen defect based on co-attention[J]. Journal of Electronics & Information Technology, 44, 1476-1483(2022).

    [8] Zhao F, Zhong B B, Liu H Q. Multi-scale residual U-Net fundus blood vessel segmentation based on attention mechanism[J]. Laser & Optoelectronics Progress, 59, 1810002(2022).

    [9] Zhang X L, Wei Y C, Yang Y et al. SG-one: similarity guidance network for one-shot semantic segmentation[J]. IEEE Transactions on Cybernetics, 50, 3855-3865(2020).

    [10] Dong Y, Pan H W, Cui Q N et al. Few-shot segmentation method for multi-modal magnetic resonance images of brain tumor[J]. Journal of Computer Applications, 41, 1049-1054(2021).

    [11] Wang J J, Zhang J H. Small sample remote sensing image segmentation based on multiscale feature fusion[J]. Journal of Huazhong University of Science and Technology (Natural Science Edition), 50, 62-67(2022).

    [12] Khan R A, Luo Y G, Wu F X. RMS-UNet: residual multi-scale UNet for liver and lesion segmentation[J]. Artificial Intelligence in Medicine, 124, 102231(2022).

    [13] Yin S C, Deng H M, Xu Z L et al. SD-UNet: a novel segmentation framework for CT images of lung infections[J]. Electronics, 11, 130(2022).

    [14] Yang Z X, Han J N, Yu B Z et al. Improved medical image segmentation method based on DeepLab V3+[J]. Foreign Electronic Measurement Technology, 40, 18-23(2021).

    [15] Dai J F, Qi H Z, Xiong Y W et al. Deformable convolutional networks[C], 764-773(2017).

    [16] Zhao Y, Zhao L J, Kuang G Y. Attention feature fusion network for rapid aircraft detection in SAR images[J]. Acta Electronica Sinica, 49, 1665-1674(2021).

    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: Xiaojie Gong (1692994031@qq.com)

    DOI:10.3788/LOP221756

    Topics