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

    An improved U2-Net segmentation model was proposed to address the issues of weak feature extraction ability and low segmentation accuracy in metal coating defect image segmentation. Firstly, an improved receptive field block light (RFB_l) module was embedded in the U-shaped residual block (RSU) to form a new feature extraction layer, which enhanced the ability to learn detailed features and solved the problem of low segmentation accuracy caused by limited receptive field in the network. Secondly, in the decoding stage of the U2-Net segmentation model, an effective contour enhanced attention (CEA) mechanism was introduced to suppress redundant features in the network, obtain feature attention maps with detailed position information, enhance the difference between boundary and background information, and achieve the more accurate segmentation results. The experimental results show that the average intersection and union ratio, accuracy, precision, recall, and F1-measure of the model on two metal coating peeling and corrosion datasets are 80.36%, 96.29%, 87.43%, 84.61%, and 86.00%, respectively. Compared with commonly used SegNet, U-Net, and U2-Net segmentation networks, the performance of the model is significantly improved.

    Keywords
    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: ZHU Daixian (朱代先)

    DOI:10.5768/JAO202445.0402005

    Topics