Open Access
Subscription Access
Lightweight CNN Models for Product Defect Detection with Edge Computing in Manufacturing Industries
Detecting product defects is one of the manufacturing industry's most essential processes in quality control. Human visual inspection for product defects is the traditional method employed in the industry. Nevertheless, it can be laborious, prone to human mistakes, and unreliable. Deep Learning-based Convolution Neural Networks (CNN) has been extensively used in fully automating product defect detection systems. However, real-time edge devices installed at the manufacturing site generally have limited computing capability and cannot run different CNN models. A lightweight CNN model is adopted in this scenario to find a balance between defect detection, model training time, memory consumption, computing time and efficiency. This work provides lightweight CNN models with transfer learning for product defect detection on fabric, surface, and casting datasets. We deployed the trained model to the NVIDIA Jetson Nano-kit edge device for detection speed with better simulation results in terms of accuracy, sensitivity rate, specificity rate, and F1 measure in the workplace context of the Manufacturing Industries.
Keywords
Convolution Neural Network, Deep Learning, Edge Devices, Lightweight Model.
User
Font Size
Information
- Routray S K, Sharmila K P, Javali A, Ghosh A D & Sarangi S, An outlook of narrowband IoT for industry 4.0, In 2020 2nd Int ConfInv Res Comput Appl (IEEE) 2020, 923–926, DOI: 10.1109/ICIRCA48905.2020.9182803
- Kalyani G, Janakiramaiah B, Prasad L V, Karuna A & Babu A M, Efficient crowd counting model using feature pyramid network and ResNeXt, Soft Comput, 25(15) (2021) 10497–10507, https://doi.org/10.1007/s00500-021-05993-x
- Subramanian M, Lv N P & VE S, Hyperparameter optimization for transfer learning of VGG16 for disease identification in corn leaves using Bayesian optimization, Big Data, 10(3) (2022) 215–229, DOI: 10.1089/big.2021.0218.
- 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, DOI: 10.3233/JIFS-210593
- Grigorescu S, Trasnea B, Cocias T & Macesanu G, A survey of deep learning techniques for autonomous driving, J Field Robotics, 37(3) (2020) 362–386.
- Konovalenko I, Maruschak P, Brevus V & Prentkovskis O, Recognition of scratches and abrasions on metal surfaces using a classifier based on a convolutional neural network, Metals, 11(4) (2021) 549, https://doi.org/10.3390/met11040549
- Yu J, Zhang D, Li H, Song C, Zhou X, Shen S, ... & Wang H, Detection of internal holes in additive manufactured Ti-6Al-4V part using laser ultrasonic testing, Appl Sci, 10(1) (2020) 365.
- Pan Z, Yang J, Wang X E, Wang F, Azim I & Wang C, Image-based surface scratch detection on architectural glass panels using deep learning approach, Constr Build Mater, 282 (2021) 122717, https://doi.org/10.1016/j.conbuildmat.2021.122717
- Sahoo S K, Sharma M M & Choudhury B B, A dynamic bottle inspection structure, In Comput Intell Data Mining (Springer) 2019, 873–884.
- Ren R, Hung T & Tan K C, A generic deep-learning-based approach for automated surface inspection, IEEE Trans Cybernetics, 48(3) (2017) 929–940, doi: 10.1109/TCYB.2017.2668395.
- Jiang J, Cao P, Lu Z, Lou W & Yang Y, Surface defect detection for mobile phone back glass based on symmetric convolutional neural network deep learning, Appl Sci, 10(10) (2020) 3621.
- Hu C & Wang Y, An efficient convolutional neural network model based on object-level attention mechanism for casting defect detection on radiography images, IEEE Trans Ind Electron, 67(12) (2020) 10922–10930.
- Ng H F, Automatic thresholding for defect detection, Patt Recogn Lett, 27(14) (2006) 1644–1649.
- Bai X, Fang Y, Lin W, Wang L & Ju B F, Saliency-based defect detection in industrial images by using phase spectrum, IEEE TransInd Inf, 10(4) (2014) 2135–2145.
- Duan L, Yang K & Ruan L, Research on automatic recognition of casting defects based on deep learning, IEEE Access, 9 (2020) 12209–12216.
- Lin J, Yao Y, Ma L & Wang Y, Detection of a casting defect tracked by deep convolution neural network, Int J Adv Manuf Technol, 97(1) (2018) 573–581.
- Song S, Jing J, Huang Y & Shi M, EfficientDet for fabric defect detection based on edge computing, J Eng Fibers Fabr, 16 (2021) 15589250211008346.
- Srinivasan K, Dastoor P H, Radhakrishnaiah P & Jayaraman S, FDAS: a knowledge-based framework for analysis of defects in woven textile structures, J Textile Ins, 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 Electric Electron Comput Sci, BiomedEng' Meet (IEEE) 2018, 1–5.
- Jolliffe I T & Cadima J, Principal component analysis: a review and recent developments, Philo 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.
- He Y, Huang X Y & Tay F E H, Fabric defect detection based on object as point, In CS IT Conf Proc, 6(11) (2021).
- Peng Z, Gong X, Wei B, Xu X & Meng S, Automatic unsupervised fabric defect detection based on self-feature comparison, Electron, 10(21) (2021) 2652.
- Sandler M, Howard A, Zhu M, Zhmoginov A & Chen L C, Mobilenetv2: Inverted residuals and linear bottlenecks, In Proc IEEE Conf Comput Vis Patt Recogn (IEEE) 2018, 4510–4520.
- Iandola F N, Han S, Moskewicz M W, Ashraf K, Dally W J & Keutzer K, SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and < 0.5 MB model size. arXiv preprint arXiv:1602.07360 (2016).
- Zhang X, Zhou X, Lin M. Sun J, ShuffleNet: An extremely efficient convolutional neural network for mobile devices, in Proc 2018 IEEE/CVF Conf Comput Vis Patt Recogn (IEEE) 2018, 6848–6856.
- Huang G, Liu S, van der Maaten L & Weinberger K Q, CondenseNet: An efficient densenet using learned group convolutions, in Proc 2018 IEEE/CVF Conf Comput Vis Patt Recogn (IEEE) 2018, 2752–2761.
- Wang R J, Li X & Ling C X, Pelee: A real-time object detection system on mobile devices, Adv in Neural Infoproc Syst, 31 (2018).
- Qin X & Wang Z, Nasnet: A neuron attention stage-by-stage net for single image deraining, arXiv preprint arXiv: 1912.03151 (2019).
- Seker A, Peker K A, Yüksek A G & Delibas E, Fabric defect detection using deep learning, In Proc the 24th Sig Proc Commut Appl Conf (IEEE) 2016, 1437–1440.
- Schlagenhauf T, Ball Screw Drive Surface Defect Dataset for Classification (2021)
- Dabhi R, Casting product image data for quality inspection, (Kaggle.com) 2020.
Abstract Views: 134
PDF Views: 83