Loading [a11y]/accessibility-menu.js
Cam-Guided U-Net With Adversarial Regularization For Defect Segmentation | IEEE Conference Publication | IEEE Xplore

Cam-Guided U-Net With Adversarial Regularization For Defect Segmentation


Abstract:

Defect segmentation is critical in real-wold industrial product quality assessment. There are usually a huge number of normal (defect-free) images but a very limited numb...Show More

Abstract:

Defect segmentation is critical in real-wold industrial product quality assessment. There are usually a huge number of normal (defect-free) images but a very limited number of annotated anomalous images. This poses huge challenges to exploiting Fully-Convolutional Networks (FCN), e.g., UNet, as they require sufficient anomalous images with defect annotations during training. To further leverage the information from normal data, a novel CAM-guided U-Net with adversarial regularization (CAM-UNet-AR) is proposed. We first modify the existing CAM-UNet to incorporate the CAMs for both normal and anomalous classes and fine-tune the segmentation network using a combined loss which jointly considers pixel-wise classification, foreground segmentation and boundary segmentation. Secondly, an auxiliary adversarial regularization module (ARM) is proposed to facilitate the segmentation network to encode the “normal components” from training images into consistent representations. Extensive experiments on MVTec AD dataset show the superiority of our proposed network over multiple state-of-the-art U-Net variants.
Date of Conference: 19-22 September 2021
Date Added to IEEE Xplore: 23 August 2021
ISBN Information:

ISSN Information:

Conference Location: Anchorage, AK, USA

Contact IEEE to Subscribe

References

References is not available for this document.