Efficient Object Detection and Intelligent Information Display Using YOLOv4-Tiny

Ying-Tung Hsiao; Jia-Shing Sheu; Hsu Ma

doi:10.46604/aiti.2023.12682

Authors

Ying-Tung Hsiao Department of Computer Science, National Taipei University of Education, Taipei, Taiwan, ROC
Jia-Shing Sheu Department of Computer Science, National Taipei University of Education, Taipei, Taiwan, ROC
Hsu Ma Department of Computer Science, National Taipei University of Education, Taipei, Taiwan, ROC

DOI:

https://doi.org/10.46604/aiti.2023.12682

Keywords:

YOLOv4-tiny, deep learning, object detection, image recognition, information display

Abstract

This study aims to develop an innovative image recognition and information display approach based on you only look once version 4 (YOLOv4)-tiny framework. The lightweight YOLOv4-tiny model is modified by replacing convolutional modules with Fire modules to further reduce its parameters. Performance reductions are offset by including spatial pyramid pooling, and they also improve the model’s detection ability for objects of various sizes. The pattern analysis, statistical modeling, and computational learning visual object classes (PASCAL VOC) 2012 dataset are used, the proposed modified YOLOv4-tiny architecture achieves a higher mean average precision (mAP) that is 1.59% higher than its unmodified counterpart. This study addresses the need for efficient object detection and recognition on resource-constrained devices by leveraging YOLOv4-tiny, Fire modules, and SPP to achieve accurate image recognition at a low computational cost.

References

M. S. B. Hossain, J. Dranetz, H. Choi, and Z. Guo, “DeepBBWAE-Net: A CNN-RNN Based Deep SuperLearner for Estimating Lower Extremity Sagittal Plane Joint Kinematics Using Shoe-Mounted IMU Sensors in Daily Living,” IEEE Journal of Biomedical and Health Informatics, vol. 26, no. 8, pp. 3906-3917, August 2022.

S. S. Islam, E. K. Dey, M. N. A. Tawhid, and B. M. M. Hossain, “A CNN Based Approach for Garments Texture Design Classification,” Advances in Technology Innovation, vol. 2, no. 4, pp. 119-125, October 2017.

S. Ren, K. He, R. Girshick, and J. Sun, “Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 39, no. 6, pp. 1137-1149, June 2017.

W. Wu, Y. Yin, X. Wang, and D. Xu, “Face Detection with Different Scales Based on Faster R-CNN,” IEEE Transactions on Cybernetics, vol. 49, no. 11, pp. 4017-4028, November 2019.

X. Bi, J. Hu, B. Xiao, W. Li, and X. Gao, “IEMask R-CNN: Information-Enhanced Mask R-CNN,” IEEE Transactions on Big Data, vol. 9, no. 2, pp. 688-700, April 2023.

W. Liu, D. Anguelov, D. Erhan, C. Szegedy, S. Reed, C. Y. Fu, et al., “SSD: Single Shot MultiBox Detector,” Computer Vision – ECCV 2016: 14th European Conference on Computer Vision, pp. 21-37, October 2016.

K. H. Tseng, M. Y. Chung, W. H. Jhou, W. J. Chang, and C. H. Xie, “Development of Non-Contact Real-Time Monitoring System for Animal Body Temperature,” Proceedings of Engineering and Technology Innovation, vol. 21, pp. 27-33, April 2022.

D. Zhu, G. Xu, J. Zhou, E. Di, and M. Li, “Object Detection in Complex Road Scenarios: Improved YOLOv4-Tiny Algorithm,” 2nd Information Communication Technologies Conference, pp. 75-80, May 2021.

M. B. Ullah, “CPU Based YOLO: A Real Time Object Detection Algorithm,” IEEE Region 10 Symposium (TENSYMP), pp. 552-555, June 2020.

S. Chen and W. Lin, “Embedded System Real-Time Vehicle Detection Based on Improved YOLO Network,” 3rd Advanced Information Management, Communicates, Electronic and Automation Control Conference, pp. 1400-1403, October 2019.

C. Y. Wang, A. Bochkovskiy, and H. Y. M. Liao, “Scaled-YOLOv4: Scaling Cross Stage Partial Network,” IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 13029-13038, June 2021.

F. N. Iandola, S. Han, M. W. Moskewicz, K. Ashraf, W. J. Dally, and K. Keutzer, “SqueezeNet: AlexNet-Level Accuracy with 50x Fewer Parameters and <0.5 MB Model Size,” https://doi.org/10.48550/arXiv.1602.07360, November 04, 2016.

Z. Huang, J. Wang, X. Fu, T. Yu, Y. Guo, and R. Wang, “DC-SPP-YOLO: Dense Connection and Spatial Pyramid Pooling Based YOLO for Object Detection,” Information Sciences, vol. 522, pp. 241-258, June 2020.

J. Redmon, S. Divvala, R. Girshick, and A. Farhadi, “You Only Look Once: Unified, Real-Time Object Detection,” IEEE Conference on Computer Vision and Pattern Recognition, pp. 779-788, June 2016.

J. Redmon and A. Farhadi, “YOLO9000: Better, Faster, Stronger,” IEEE Conference on Computer Vision and Pattern Recognition, pp. 6517-6525, July 2017.

J. Redmon and A. Farhadi, “YOLOv3: An Incremental Improvement”, https://doi.org/10.48550/arXiv.1804.02767, April 08, 2018.

A. Bochkovskiy, C. Y. Wang, and H. Y. M. Liao. “YOLOv4: Optimal Speed and Accuracy of Object Detection,” https://doi.org/10.48550/arXiv.2004.10934, April 23, 2020.

C. Y. Wang, H. Y. M. Liao, Y. H. Wu, P. Y. Chen, J. W. Hsieh, and I. H. Yeh, “CSPNet: A New Backbone that can Enhance Learning Capability of CNN,” IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, pp. 1571-1580, June 2020.

K. He, X. Zhang, S. Ren, and J. Sun, “Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 37, no. 9, pp. 1904-1916, September 2015.

W. Fang, L. Wang, and P. Ren, “Tinier-YOLO: A Real-Time Object Detection Method for Constrained Environments,” IEEE Access, vol. 8, pp. 1935-1944, 2020.

E. Prasetyo, N. Suciati, and C. Fatichah, “YOLOv4-Tiny and Spatial Pyramid Pooling for Detecting Head and Tail of Fish,” International Conference on Artificial Intelligence and Computer Science Technology, pp. 157-161, June 2021.

J. S. Sheu and C. Y. Han, “Combining Cloud Computing and Artificial Intelligence Scene Recognition in Real-Time Environment Image Planning Walkable Area,” Advances in Technology Innovation, vol. 5 no. 1, pp. 10-17, January 2020.