Journal of Applied Optics, Volume. 45, Issue 4, 759(2024)

Image segmentation method for metal coating peeling and corrosion based on improved U2-Net network

Yunfeng NI1, Qingting QI1, Daixian ZHU1、*, Qiang QIU1, and Shulin LIU2
Author Affiliations
  • 1College of Communication and Information Engineering, Xi'an University of Science and Technology, Xi'an 710054, China
  • 2College of Electrical and Control Engineering, Xi'an University of Science and Technology, Xi'an 710054, China
  • show less
    Figures & Tables(13)
    Structure diagram of U2-Net network
    Structure diagram of U-shaped residual block
    Structure diagram of RFB network
    Structure diagram of RFB_l network
    RSU block with enlarged receptive field
    Edge-enhanced attention module
    Structure diagram of CEA-U2-RFB_lNet network
    Images of metal coating peeling and corrosion
    Comparison images of segmentation results
    • Table 1. Experiment-related configuration

      View table
      View in Article

      Table 1. Experiment-related configuration

      名称数值
      学习率0.001
      正则化系数0.9
      epoch100
      CUDA版本10.0
      批处理大小10
      优化器Adam
    • Table 2. Experimental results of different networks on metal coating peeling dataset %

      View table
      View in Article

      Table 2. Experimental results of different networks on metal coating peeling dataset %

      网络交并比准确率查准率召回率F1
      SegNet[23]70.7894.7173.0966.5069.64
      U-Net[16]69.1192.0678.8375.3277.04
      DeepLab[5]74.3293.8781.4578.7980.10
      U2-Net[15]77.9795.7385.0981.7683.39
      本文方法81.7797.0788.3485.9187.11
    • Table 3. Experimental results of different networks on metal coating corrosion dataset %

      View table
      View in Article

      Table 3. Experimental results of different networks on metal coating corrosion dataset %

      网络交并比准确率查准率召回率F1
      SegNet[23]67.1391.3271.2663.5467.19
      U-Net[16]63.5390.4976.9773.3575.12
      DeepLab[5]69.1691.9879.3175.1777.18
      U2-Net[15]74.2193.7383.4378.9681.13
      本文方法78.9495.5186.5283.3184.88
    • Table 4. Comparison of ablation experiment results

      View table
      View in Article

      Table 4. Comparison of ablation experiment results

      网络RFB-RSURFB_l-RSUCEA交并比/%准确率/%查准率/%召回率/%F1/%测试时间/s
      原算法×××74.2193.7383.4378.9681.1326.91
      算法1××75.8393.9784.0779.7381.8427.44
      算法2××76.0294.1384.9480.2182.5127.16
      算法3××76.9193.9284.5181.0982.7627.53
      本文方法×78.9495.5186.5283.3184.8828.11
    Tools

    Get Citation

    Copy Citation Text

    Yunfeng NI, Qingting QI, Daixian ZHU, Qiang QIU, Shulin LIU. Image segmentation method for metal coating peeling and corrosion based on improved U2-Net network[J]. Journal of Applied Optics, 2024, 45(4): 759

    Download Citation

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

    Category: Research Articles

    Received: Jun. 21, 2023

    Accepted: --

    Published Online: Oct. 21, 2024

    The Author Email: Daixian ZHU (朱代先)

    DOI:10.5768/JAO202445.0402005

    Topics