基于深度特征修复重构的工业缺陷检测

INDUSTRIALDEFECTDETECTIONBASEDONDEEP FEATURE REPAIRAND RECONSTRUCTION

  • 摘要: 针对现实工业生产场景中收集缺陷样本困难等问题,提出一种基于改进生成对抗网络修复深度特征的缺陷检测方法。设计缺陷拟合模块同时加入预测损失函数,用以提高检测精度。利用预训练模型及特征融合模块充分挖掘输入图像的深度特征。设计改造生成对抗网络中生成器的跳连接结构,同时将鉴别器中融入自注意力机制,以提高模型修复深度特征的能力。模型在MVTecAD工业数据集上进行实验验证,平均AUC值达到0.964,比次优模型提高了2.3百分点。模型在自制铸件工业数据集上检测,平均AUC值达到了0.988,平均检测速度为260ms/幅。实验证明所提方法适用于现实工业生产场景。

     

    Abstract: In response to the difficulties in collecting defect samples in real- world industrial production scenarios, this paper proposes a defect detection model based on improved generative adversarial networks and deep feature repair. This paper designed a defect fitting module and incorporated a prediction loss function to improve detection accuracy. The pre- trained model and the feature fusion module were used to fully mine the depth features of input images. This paper designed and modified the skip connection structure of generators in GANs, while incorporating self attention mechanism into the discriminator to improve the model's ability to repair deep features. Experimental results on the MVTec AD dataset show that the average AUC reached 0.964. Compared with the sub- optimal model, it was increased by 2.3 percentage points. Experimental results on the self- made casting industry dataset show that the average AUC reached 0.988 and the average detection speed was 260ms per piece. It verifies that the proposed model is suitable for real- world industrial production scenarios.

     

/

返回文章
返回