Fabric defect detection under complex illumination based on an improved recurrent attention model

Abstract

To solve the problem of fabric defect detection under complex illumination conditions, the Recurrent Attention Model (RAM) which is insensitive to illumination and noise differences has been introduced. However, the policy gradient algorithm in the RAM has some problems, such as the difficulty of convergence and the inefficiency of the algorithm due to the shortcomings of round updating. In this paper, the Deep Deterministic Policy Gradient- Recurrent Attention Model (DDPG-RAM) algorithm is proposed to solve the problems of policy gradient algorithm. Although the decoupling of the reinforcement learning task and classification task will lead to the inconsistency of the data, the gradient variance will be smaller, and the convergence speed and stability will be accelerated. Experiment results show that fabric defects can be detected by the proposed DDPG-RAM algorithm under complex illumination conditions. Compared with RAM and the Convolutional Neural Network (CNN), the accuracy of the decoupled algorithm is 95.24%, and the convergence speed is 50% faster than that of the RAM.