Soft matching network with application to defect inspection

作者:

Highlights:

摘要

Defect detection with templates is a major concern in manufacturing. Including subtraction and template matching, traditional methods based on images stumbled over the diverse disparities of input image pair. Yet, learning-based approaches have not been explored on this task. This paper proposed a learning-based soft template matching network for defect detection, using an innovative attention mechanism.Employing feature-pyramid-network-based atrous convolution enables our model to perceive multi-scale features. The proposed contrastive attention module enhances the query feature map. Experimental results demonstrate that our network can capture defects under the interference of disparities based on the correspondence of input image pair, showing practical value for industrial defect detection.

论文关键词:Attention mechanism,Convolutional neural network,Defect detection,Soft template,Feature coupling

论文评审过程:Received 4 September 2020, Revised 11 April 2021, Accepted 12 April 2021, Available online 28 April 2021, Version of Record 6 May 2021.

论文官网地址:https://doi.org/10.1016/j.knosys.2021.107045