• Journal of Applied Optics
  • Vol. 45, Issue 4, 759 (2024)
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
    DOI: 10.5768/JAO202445.0402005 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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