Browse > Article

Development of a transfer learning based detection system for burr image of injection molded products  

Yang, Dong-Cheol (SimTech)
Kim, Jong-Sun (Shape Manufacturing R&D Department, Korea Institute of Industrial Technology)
Publication Information
Design & Manufacturing / v.15, no.3, 2021 , pp. 1-6 More about this Journal
Abstract
An artificial neural network model based on a deep learning algorithm is known to be more accurate than humans in image classification, but there is still a limit in the sense that there needs to be a lot of training data that can be called big data. Therefore, various techniques are being studied to build an artificial neural network model with high precision, even with small data. The transfer learning technique is assessed as an excellent alternative. As a result, the purpose of this study is to develop an artificial neural network system that can classify burr images of light guide plate products with 99% accuracy using transfer learning technique. Specifically, for the light guide plate product, 150 images of the normal product and the burr were taken at various angles, heights, positions, etc., respectively. Then, after the preprocessing of images such as thresholding and image augmentation, for a total of 3,300 images were generated. 2,970 images were separated for training, while the remaining 330 images were separated for model accuracy testing. For the transfer learning, a base model was developed using the NASNet-Large model that pre-trained 14 million ImageNet data. According to the final model accuracy test, the 99% accuracy in the image classification for training and test images was confirmed. Consequently, based on the results of this study, it is expected to help develop an integrated AI production management system by training not only the burr but also various defective images.
Keywords
Artificial neural network; Burr; Image; Transfer learning;
Citations & Related Records
연도 인용수 순위
  • Reference
1 LeCun, Y., Bottou, L., Bengio, Y., and Haffner, P. "Gradient-based learning applied to document recognition", Proc. IEEE Vol. 86, No. 11, pp. 2278-2324, 1998.   DOI
2 Torrey, L. and Shavlik, J. "Transfer Learning", IGI global, Pennsylvania, pp. 242-264, 2010.
3 Otsu, N. "A threshold selection method from gray-level histograms", IEEE Trans. Syst, man Cybern. Vol. 9, No. 1, pp. 62-66, 1979.
4 Yosinski, J., Clune, J., Bengio, Y., and Lipson, H. "How transferable are features in deep neural networks?", arXiv preprint arXiv:1411.1792, 2014.
5 Zoph, B., Vasudevan, V., Shlens, J., and Le, Q. V. "Transfer learning transferable architectures for scalable image recognition", 2018 IEEE/CVF Conf. Comput. Vision Pattern Recog. pp. 8697-8710. 2018.
6 Example of ImageNet data set, https://www.image-net.org.
7 Latorella, K. A. and Prabhu, P. V., "A review of human error in aviation maintenance and inspection", Int. J. Ind. Ergon. Vol. 26, No. 2, pp. 133-161, 2000.   DOI
8 Bruning, J. H., Feldman, M., Kinsel, T. S., Sittig, E. K., and Townsend, R. L. "An automated mask inspection system AMIS", IEEE Trans. Electron Devices Vol. 22, No. 7, pp. 487-495, 1975.   DOI