.png){kind=icon}
返回
Defect attention template generation cycleGAN for weakly supervised surface defect segmentation
.png){kind=icon}
PRE
AI
DOI：10.1016/j.patcog.2021.108396.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Surface defect segmentation is very important for the quality inspection of industrial production and is an important pattern recognition problem. Although deep learning (DL) has achieved remarkable results in surface defect segmentation, most of these results have been obtained by using massive images with pixel-level annotations, which are difficult to obtain at industrial sites. This paper proposes a weakly supervised defect segmentation method based on the dynamic templates generated by an improved cycle consistent generative adversarial network (CycleGAN) trained by image-level annotations. To generate better templates for defects with weak signals, we propose a defect attention module by applying the defect residual for the discriminator to strengthen the elimination of defect regions and suppress changes in the background. A defect cycle-consistent loss is designed by adding structural similarity (SSIM) to the original L1 loss to include the grayscale and structural features; the proposed loss can better model the inner structure of defects. After obtaining the defect-free template, a defect segmentation map can easily be obtained through a simple image comparison and threshold segmentation. Experiments show that the proposed method is both efficient and effective, significantly outperforms other weakly supervised methods, and achieves performance that is comparable or even superior to that of supervised methods on three industrial datasets (intersection over union (IoU) on the DAGM 2007, KSD and CCSD datasets of 78.28%, 59.43%,and 68.83%, respectively). The proposed method can also be employed as a semiautomatic annotation tool combined with active learning.(c) 2021 Elsevier Ltd. All rights reserved.
Keyword:
Weakly supervised learning
Defect detection
Image segmentation
Generative adversarial network (GAN)
Attention model
Weakly supervised learning
Defect detection
Image segmentation
Generative adversarial network (GAN)
Attention model
期刊
IF:
7.6
论文数:
1.3W
被引数:
4.5W
机构
暂无机构信息
引用论文
Mixed supervision for surface-defect detection: From weakly to fully supervised learning表面缺陷检测的混合监督: 从弱监督学习到完全监督学习
OAAI
OAAI分享收藏
A High-Efficiency Fully Convolutional Networks for Pixel-Wise Surface Defect Detection.png)
IEEE ACCESSOAAI
IEEE ACCESSIF3.6
OAAI分享收藏Is SA shared or distributed in team work? An exploratory study in an intelligence analysis task
PREAI
分享收藏CADN: A weakly supervised learning-based category-aware object detection network for surface defect detectionCADN: 一种基于弱监督学习的类别感知物体检测网络，用于表面缺陷检测PATTERN RECOGNITION
PATTERN RECOGNITIONIF7.6
PREAI
分享收藏
查看更多内容