Journal of Scientific and Industrial Research

Open Access Open Access  Restricted Access Subscription Access

MobileNetV2-Based Transfer Learning Model with Edge Computing for Automatic Fabric Defect Detection


Affiliations
1 Computer Science and Engineering, Prasad V. Potluri Siddhartha Institute of Technology, Vijayawada 520 007, Andhra Pradesh, India
2 University College of Engineering Kakinada (A), Jawaharlal Nehru Technological University Kakinada 533 003, Andhra Pradesh, India
3 Computer Science and Engineering, Seshadri Rao Gudlavalleru Engineering College, Gudlavalleru 521 356, Andhra Pradesh, India
4 Computer Science and Engineering, Sir C R Reddy College of Engineering, Eluru 534 007, Andhra Pradesh, India
5 Computer Science and Engineering, Koneru Lakshmaiah Education Foundation, Vaddeswaram 522 302, Andhra Pradesh, India
 

In textile manufacturing, fabric defect detection is an essential quality control step and a challenging task. Earlier, manual efforts were applied to detect defects in fabric production. Human exhaustion, time consumption, and lack of concentration are the main problems in the manual defect detection process. Machine vision systems based on deep learning play a vital role in the Industrial Internet of things (IIoT) and fully automated production processes. Deep learning centered on Convolution Neural Network (CNN) models have been commonly used in fabric defect detection, but most of these models require high computing resources. This work presents a lightweight MobileNetV2-based Transfer Learning model to assist defect detection with low power consumption, low latency, easy upgrade, more efficiency, and an automatic visual inspection system with edge computing. Firstly, different image transformation techniques were performed as data augmentation on four fabric datasets for the model's adaptability in various fabrics. Secondly, fine-tuning hyperparameters of the MobileNetV2 with transfer learning gives a lightweight, adaptable and scalable model that suits the resource-constrained edge device. Finally, deploy the trained model to the NVIDIA Jetson Nano-kit edge device to make its detection faster. We assessed the model based on its accuracy, sensitivity rate, specificity rate, and F1 measure. The numerical simulation reveals that the model accuracy is 96.52%, precision is 96.52%, recall is 96.75%, and F1-Score is 96.52%.

Keywords

Deep Learning, Edge Devices, Industrial IoT, Modeling, MobileNetV2.
User
Notifications
Font Size

  • Najafabadi F S & Pourghassem H, Corner defect detection based on dot product in ceramic tile images, in 2011 IEEE 7th Int Colloq Signal Process Appl (UniversitiTeknologiMARA) 2011, 293–297.

  • Elham H, Farhadi F & Tajeripour F, Fabric defect detection using auto-correlation function, Int J Comput Theory, 5(1) (2013) 114.

  • Ghazvini M, Monadjemi S A, Movahhedinia N & Jamshidi K, Defect detection of tiles using 2D-wavelet transform and statistical features, World Acad Sci Eng Technol, 49 (2009) 901–904.

  • Janakiramaiah B, Kalyani G & Jayalakshmi A, Automatic alert generation in a surveillance system for smart city environment using deep learning algorithm, Evol Intell, 14(2) (2021) 635–642.

  • Zhou X, Li Y & Liang W, CNN-RNN based intelligent recommendation for online medical pre-diagnosis support, IEEE/ACM Trans Comput Biol Bioinform 18(3) (2020) 912–921.

  • Rasheed A, Zafar B, Rasheed A, Ali N, Sajid M, Dar S H, Habib U, Shehryar T & Mahmood M T, Fabric defect detection using computer vision techniques: a comprehensive review, Math Probl Eng (2020). https://doi.org/10.1155/2020/8189403

  • Srinivasan K, Dastoor P H, Radhakrishnaiah P & Jayaraman S, FDAS: a knowledge-based framework for analysis of defects in woven textile structures, J Text Inst, 83(3) (1992) 431–448.

  • Chan C H & Pang G K, Fabric defect detection by Fourier analysis, IEEE Trans Ind Appl, 36(5) (2000) 1267–1276.

  • Selvi S S T & Nasira G M, An effective automatic fabric defect detection system using digital image processing, J Environ Nanotechnol, 6(1) (2017) 79–85.

  • Sadaghiyanfam S, Using gray-level-co-occurrence matrix and wavelet transform for textural fabric defect detection: A comparison study, in 2018 Elect Electrn Comput Sci Biomedi Eng Meeting (EBBT) (IEEE) 2018, 1–5.

  • Anandan P & Sabeenian R S, Fabric defect detection using discrete curvelet transform, Procedia Comput Sci, 133 (2018) 1056–1065.

  • Bertolini M, Mezzogori D, Neroni M & Zammori F, Machine Learning for industrial applications: A comprehensive literature review, Expert Syst Appl, 175 (2021) 114820. https://doi.org/10.1016/j.eswa.2021.114820

  • Jolliffe I T & Cadima J, Principal component analysis: a review and recent developments, Philos Trans Royal Soc A: Math Phy Eng Sci, 374(2065) (2016) 20150202.

  • Li M, Wan S, Deng Z & Wang Y, Fabric defect detection based on saliency histogram features, Comput Intell, 35(3) (2019) 517–534.

  • Mei S, Wang Y & Wen G, Automatic fabric defect detection with a multi-scale convolutional denoising autoencoder network model, Sensors 18(4) (2018)1064.

  • Zhang W, Jia M P, Zhu L & Yan X A, Comprehensive overview on computational intelligence techniques for machinery condition monitoring and fault diagnosis, Chinese J Mech Eng, 30(4) (2017) 782–795.

  • Khan A, Sohail A, Zahoora U & Qureshi A S, A survey of the recent architectures of deep convolutional neural networks. Artif Intell Rev 53(8) (2020) 5455–5516.

  • Alzubaidi L, Zhang J, Humaidi A J, Al-Dujaili A, Duan Y, Al-Shamma O & Farhan L, Review of deep learning: Concepts, CNN architectures, challenges, applications, future directions, J Big Data, 8(1) (2021) 1–74.

  • Beljadid A, Tannouche A & Balouki A, Application of deep learning for the detection of default in fabric texture, in 2020 IEEE 6th Int Conf Optim Appl, (IEEE) 2020, 1–5.

  • Janakiramaiah B, Kalyani G, Karuna A, Prasad L V & Krishna M, Military object detection in defense using multi-level capsule networks, Soft Comput (2021) 1–5. https://doi.org/10.1007/s00500-021-05912-0

  • Janakiramaiah B, Kalyani G, Prasad L V, Karuna A & Krishna M, Intelligent system for leaf disease detection using capsule networks for horticulture, J Intell Fuzzy Syst, 41(6) (2021) 6697–6713.

  • He Y, Huang X Y & Tay F E H, Fabric defect detection based on object as point, in CS & IT Conf Proc ONEDU, 11(6) (2021) 99–107. DOI: 10.5121/csit.2021.110608

  • Peng Z, Gong X, Wei B, Xu X & Meng S, Automatic unsupervised fabric defect detection based on self-feature comparison, Electronics, 10(21) (2021) 2652.

  • Li E, Zeng L, Zhou Z & Chen X, Edge AI: On-demand accelerating deep neural network inference via edge computing, IEEE Trans Wirel Commun, 19(1) (2019) 447–457.

  • Shi W, Sun H, Cao J, Zhang Q & Liu W, Edge computing-an emerging computing model for the internet of everything era, J Comput Res Dev, 54(5) (2017) 907–924,

  • Song S, Jing J, Huang Y & Shi M, EfficientDet for fabric defect detection based on edge computing, J Eng Fibers Fabr, 16 (2021) 15589250211008346.

  • Zhu Z, Han G, Jia G & Shu L, Modified densenet for automatic fabric defect detection with edge computing for minimizing latency, IEEE Internet Things J, 7(10) (2020) 9623–9636.

  • Şeker A, Peker K A, Yüksek A G & Delibaş E, Fabric defect detection using deep learning, in 2016 24th Signal Process Commu Appl Conf, (SIU) (IEEE) 2016, 1437–1440 D. F. Germany, “Tilda Textile Texture Database.” [http://lmb.informatik.uni-freiburg.de/resources/datasets/ tilda.en.html. Version 1.0, 1996 (access)

  • Zhao Y, Hao K, He H, Tang X, & Wei B, A visual long-short-term memory based integrated CNN model for fabric defect image classification, Neurocomputing, 380 (2020) 259–270.

  • Sandler M, Howard A, Zhu M, Zhmoginov A, Chen L C, MobileNetV2: Inverted Residuals and Linear Bottlenecks, in Proc IEEE Conf Comput Vis Pattern Recognit, 18(22) (2018) 4510–4520.


Abstract Views: 47

PDF Views: 52




  • MobileNetV2-Based Transfer Learning Model with Edge Computing for Automatic Fabric Defect Detection

Abstract Views: 47  |  PDF Views: 52

Authors

Lakshmi Ramani Burra
Computer Science and Engineering, Prasad V. Potluri Siddhartha Institute of Technology, Vijayawada 520 007, Andhra Pradesh, India
Karuna A
University College of Engineering Kakinada (A), Jawaharlal Nehru Technological University Kakinada 533 003, Andhra Pradesh, India
Srinivasarao Tumma
Computer Science and Engineering, Seshadri Rao Gudlavalleru Engineering College, Gudlavalleru 521 356, Andhra Pradesh, India
Krishna Marlapalli
Computer Science and Engineering, Sir C R Reddy College of Engineering, Eluru 534 007, Andhra Pradesh, India
Praveen Tumuluru
Computer Science and Engineering, Koneru Lakshmaiah Education Foundation, Vaddeswaram 522 302, Andhra Pradesh, India

Abstract


In textile manufacturing, fabric defect detection is an essential quality control step and a challenging task. Earlier, manual efforts were applied to detect defects in fabric production. Human exhaustion, time consumption, and lack of concentration are the main problems in the manual defect detection process. Machine vision systems based on deep learning play a vital role in the Industrial Internet of things (IIoT) and fully automated production processes. Deep learning centered on Convolution Neural Network (CNN) models have been commonly used in fabric defect detection, but most of these models require high computing resources. This work presents a lightweight MobileNetV2-based Transfer Learning model to assist defect detection with low power consumption, low latency, easy upgrade, more efficiency, and an automatic visual inspection system with edge computing. Firstly, different image transformation techniques were performed as data augmentation on four fabric datasets for the model's adaptability in various fabrics. Secondly, fine-tuning hyperparameters of the MobileNetV2 with transfer learning gives a lightweight, adaptable and scalable model that suits the resource-constrained edge device. Finally, deploy the trained model to the NVIDIA Jetson Nano-kit edge device to make its detection faster. We assessed the model based on its accuracy, sensitivity rate, specificity rate, and F1 measure. The numerical simulation reveals that the model accuracy is 96.52%, precision is 96.52%, recall is 96.75%, and F1-Score is 96.52%.

Keywords


Deep Learning, Edge Devices, Industrial IoT, Modeling, MobileNetV2.

References