计算机工程与应用 ›› 2022, Vol. 58 ›› Issue (20): 108-116.DOI: 10.3778/j.issn.1002-8331.2111-0470

• 模式识别与人工智能 • 上一篇    下一篇

适于少样本缺陷检测的两阶段缺陷增强网络

陈朝,刘志,李恭杨,彭铁根   

  1. 1.上海大学 通信与信息工程学院,上海 200444
    2.宝山钢铁股份有限公司 中央研究院,上海 201999
  • 出版日期:2022-10-15 发布日期:2022-10-15

Two-Phase Defect Enhancement Network for Few-Shot Defect Detection

CHEN Zhao, LIU Zhi, LI Gongyang, PENG Tiegen   

  1. 1.School of Communication and Information Engineering, Shanghai University, Shanghai 200444, China
    2.Research Institute, Baoshan Iron & Steel Co.Ltd., Shanghai 201999, China
  • Online:2022-10-15 Published:2022-10-15

摘要: 缺陷检测模型一般需要大量样本来学习缺陷的特征,但实际场景中一些重要缺陷的样本难以收集,如何用少量样本来学习罕见缺陷的特征成为一个具有挑战性的问题。为了促进少样本缺陷检测的研究,构建了一个新的工业表面缺陷数据集,包括缺陷样本和无缺陷样本。同时提出了一个两阶段缺陷增强网络以提升少样本场景下的缺陷检测性能,它利用了无缺陷样本,并将整个训练过程分为两个阶段。第一阶段的训练需要大量缺陷样本,而第二阶段的训练只需要少量缺陷样本和无缺陷样本。此外,还提出了一个缺陷突显模块,可以更好地利用无缺陷样本来增强缺陷区域的特征。在新数据集上的实验表明,该缺陷检测模型的性能优于其他的少样本目标检测模型,在工业表面缺陷检测中具有更好的应用前景。

关键词: 缺陷检测, 少样本学习, 无缺陷样本, 两阶段缺陷增强网络, 缺陷突显模块

Abstract: Defect detection models generally require many training samples to learn the characteristics of defects, however, some important defects are often difficult to collect in the actual scene. How to learn the characteristics of these rare defects with few training samples becomes a challenging issue. To facilitate the study of few-shot defect detection, a new industrial surface defects dataset is constructed, including defect samples and defect-free samples. Meanwhile, a two-phase defect enhancement network is proposed to improve the performance of defect detection when only few samples are provided, which utilizes defect-free samples, and the whole training process is divided into two phases. The first phase needs many defect samples, while the second phase only needs few defect samples and defect-free samples. In addition, a defect prominence module is proposed, which makes better use of defect-free samples to enhance the characteristics of defective areas. Experiments on the new dataset show that the proposed defect detection model outperforms other few-shot object detection models and has a better application prospect in industrial surface defect detection.

Key words: defect detection, few-shot learning, defect-free samples, two-phase defect enhancement network, defect prominence module