Modification of a convolutional neural network for the weave pattern classification

The fabric quality in textile industry is characterized by the texture (weave pattern) as it plays a vital role for the production and design of best quality fabric. The earlier proposed automated weave identification methods based on image processing techniques are highly dependent on the lighting conditions. The machine learning methods have been reported to show better accuracy. However, they require very large training datasets, very high processing power and computation time. This study proposes improved accuracy with smaller dataset and reduced computation time by proposing a modification of VGG16 model by adding two additional pooling layers. Using evaluation metrics of both models, the modified model results were analysed according to accuracy, balanced accuracy, and F1-score. On the basis of investigational outcomes, a comparison has been performed with earlier work. The results show that the proposed VGG-16 model is capable to achieve state-of-the-art accuracy and avoid unnecessary activation features by freezing the main convolutional base layers. Ultimately, as evidenced by the performance of the modified VGG-16 deep learning model, the proposed method demonstrated improved accuracy. The study results show that the proposed modified VGG16 algorithm is able to recognize the features of provided database with 90% accuracy and F1-Score ranging from 0.8 to1.