DOI QR코드

DOI QR Code

Anomaly Sewing Pattern Detection for AIoT System using Deep Learning and Decision Tree

  • Received : 2023.12.19
  • Accepted : 2024.02.18
  • Published : 2024.02.29

Abstract

Artificial Intelligence of Things (AIoT), which combines AI and the Internet of Things (IoT), has recently gained popularity. Deep neural networks (DNNs) have achieved great success in many applications. Deploying complex AI models on embedded boards, nevertheless, may be challenging due to computational limitations or intelligent model complexity. This paper focuses on an AIoT-based system for smart sewing automation using edge devices. Our technique included developing a detection model and a decision tree for a sufficient testing scenario. YOLOv5 set the stage for our defective sewing stitches detection model, to detect anomalies and classify the sewing patterns. According to the experimental testing, the proposed approach achieved a perfect score with accuracy and F1score of 1.0, False Positive Rate (FPR), False Negative Rate (FNR) of 0, and a speed of 0.07 seconds with file size 2.43MB.

Keywords

Acknowledgement

This work was supported by hongik University and Ministry of SMEs and Startups

References

  1. M. R. Pedersen, L. Nalpantidis, R. S. Andersen, C. Schou, S. Bogh, V. Kruger, and O. Madsen, "Robot skills for manufacturing: From concept to industrial deployment," Robotics and Computer-Integrated Manufacturing, vol. 37, pp. 282- 291, 2016.  https://doi.org/10.1016/j.rcim.2015.04.002
  2. G. Jocher, "YOLOv5 by Ultralytics," 52020. Available at https://github.com/ultralytics/yolov5.(accessed Dec., 15, 2023). 
  3. L. Liu, W. Ouyang, X. Wang, P. Fieguth, J. Chen, X. Liu, et al., "Deep learning for generic object detection," A Survey [J]., 2018. 
  4. C. B. Murthy, M. F. Hashmi, N. D. Bokde, and Z. W. Geem, "Investigations of object detection in im- ages/videos using various deep learning techniques and embedded platforms a comprehensive review," Applied sciences, vol. 10, no. 9, pp. 3280, 2020. 
  5. X. Wu, D. Sahoo, and S. C. Hoi, "Recent advances in deep learning for object detection," Neurocomputing, vol. 396, pp. 39-64, 2020.  https://doi.org/10.1016/j.neucom.2020.01.085
  6. A. Krizhevsky, I. Sutskever, and G. E. Hinton, "Imagenet classification with deep convolutional neural networks," Communications of the ACM, vol. 60, no. 6, pp. 84-90, 2017.  https://doi.org/10.1145/3065386
  7. M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L. C. Chen, "Mobilenetv2: Inverted residuals and linear bottlenecks," in Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 4510-4520, 2018. 
  8. A. Bochkovskiy, C. Y. Wang, and H. Y. M. Liao, "Yolov4: Optimal speed and accuracy of object detection," arXiv preprint arXiv:2004.10934, 2020. 
  9. Y. Xiong, H. Liu, S. Gupta, B. Akin, G. Bender, Y. Wang, P. J. Kindermans, M. Tan, V. Singh, and B. Chen, "Mobiledets: Searching for object detection architectures for mobile accelerators," in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 3825-3834, 2021. 
  10. J. Redmon and A. Farhadi, "Yolov3: An incremental improvement," CoRR, vol. abs/1804.02767, 2018. 
  11. A. Ghosh, S. A. Al Mahmud, T. I. R. Uday, and D. M. Farid, "Assistive technology for visually impaired using tensor flow object detection in raspberry pi and coral usb accelerator," in 2020 IEEE Region 10 Symposium (TEN- SYMP), pp. 186-189, 2020. 
  12. M. Verucchi, G. Brilli, D. Sapienza, M. Verasani, M. Arena, F. Gatti, A. Capotondi, R. Cavicchioli, M. Bertogna, and M. Solieri, "A systematic assessment of embedded neural networks for object detection," in 2020 25th IEEE International Conference on Emerging Technologies and Factory Automation (ETFA), vol. 1, pp. 937- 944, IEEE, 2020. 
  13. D. N. N. Tran, H. H. Nguyen, L. H. Pham, and J. W. Jeon, "Object detection with deep learning on drive px2," in 2020 IEEE International Conference on Consumer Electronics- Asia (ICCE-Asia), pp. 1-4, IEEE, 2020. 
  14. H. H. Nguyen, D. N. N. Tran, and J. W. Jeon, "Towards real-time vehicle detection on edge devices with nvidia jetson tx2," in 2020 IEEE International Conference on Consumer Electronics - Asia (ICCE-Asia), pp. 1-4, 2020. 
  15. A. Goncalves, P. Ray, B. Soper, D. Widemann, M. Nygard, J. F. Nygard, and A. P. Sales, "Bayesian multitask learning regression for heterogeneous patient cohorts," Journal of Biomedical Informatics, vol. 100, pp. 100059, 2019. 
  16. X. Zhang, J. Zhou, W. Sun, and S. K. Jha, "A lightweight cnn based on transfer learning for covid-19 diagnosis," Computers, Materials and Continua, pp. 1123-1137, 2022. 
  17. W. Liu, D. Anguelov, D. Erhan, C. Szegedy, S. Reed, C.Y. Fu, and A. C. Berg, "SSD: Single shot multibox detector," in European conference on computer vision, pp. 21-37, Springer, 2016. 
  18. H.K. Jung and G.S. Choi, "Improved yolov5: Efficient object detection using drone images under various conditions," Applied Sciences, vol. 12, no. 14, p. 7255, 2022. 
  19. M. A. Rahaman, M. M. Ali, K. Ahmed, F. M. Bui, and S. H. Mahmud, "Performance analysis between yolov5s and yolov5m model to detect and count blood cells: deep learning approach," in Proceedings of the 2nd International Conference on Computing Advancements, pp. 316-322, 2022. 
  20. B. Jiang, R. Luo, J. Mao, T. Xiao, and Y. Jiang, "Acquisition of localization confidence for accurate object detection," in Proceedings of the European conference on computer vision (ECCV), pp. 784-799, 2018. 
  21. "Post training quantization," https://www.tensorflow.org/lite/performance/post training quantization.(accessed Dec., 15, 2023). 
  22. G. Bradski, "The opencv library.," Dr. Dobb's Journal: Software Tools for the Professional Programmer, vol. 25, no. 11, pp. 120-123, 2000. 
  23. "Libjpeg-turbo github," 07 2015. 
  24. C. R. Harris, K. J. Millman, S. J. Van Der Walt, R. Gom- mers, P. Virtanen, D. Cournapeau, E. Wieser, J. Taylor, S. Berg, N. J. Smith, et al., "Array programming with numpy," Nature, vol. 585, no. 7825, pp. 357-362, 2020.  https://doi.org/10.1038/s41586-020-2649-2
  25. Q. Wang, X. Zhang, Y. Zhang, and Q. Yi, "Augem: automatically generate high performance dense linear algebra kernels on x86 cpus," in SC'13: Proceedings of the International Conference on High Performance Computing, Networking, Storage and Analysis, pp. 1-12, IEEE, 2013.