Fabric surface detection using Small Sample Learning based on Naive Bayes

Fabric defect detection, a popular topic in automation, is a necessary and essential step of quality control in the textile manufacturing industry. Traditional machine learning algorithms such as deep learning always require a large number of training samples in fabric defect detection. However, fabric defect rate has been greatly decreased because of production technology has been developed further. An algorithm called Bayesian Small Sample Learning (BSSL) based on Naive Bayes was proposed to solve the problem of lack of training samples. Firstly, it is important to remove the noise in the image which collected from experiment platform. After that, the reference values are obtained by learning few samples of different defective fabrics and defect- free fabrics. Finally, the feature values need to be extracted from the fabric to be detected and Bayesian algorithm is used to calculate the posterior probability which the reference values to the feature values when the learning process completed. The fabric was detected as defective or not determined by maximum posterior probability. Experimental results show that the proposed algorithm BSSL requires few defective samples for learning and also can achieve high accuracy of detection.