An Efficient Application of Modified YOLOv5 in Basketball Player Detection and Analysis

Jia-Shing Sheu; Sheng-Ju Lin

doi:10.46604/aiti.2024.13702

Authors

Jia-Shing Sheu Department of Computer Science, National Taipei University of Education, Taipei, Taiwan, ROC
Sheng-Ju Lin Department of Computer Science, National Taipei University of Education, Taipei, Taiwan, ROC

DOI:

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

Keywords:

player detection, player tracking, basketball game, YOLOv5

Abstract

Effective analysis helps players evaluate their performance, make necessary adjustments, and develop diverse game strategies. Moreover, the analysis provides viewers with different perspectives, enhancing their understanding of the game. This study aims to develop a basketball player detection and analysis system to assist in analyzing on-court situations. The system uses perspective transformation to obtain player tracking information on the top view image in basketball games. The system uses a modified you only look once (YOLO) v5 model that replaces the backbone of YOLOv5s with the MobileNetv3-small architecture for player detection. Compared to the original YOLOv5, the modified YOLOv5 reduces parameters from 7.02 × 10⁶ to 3.5 × 10⁶, a decrease of 49.8%. The number of frames obtainable per second increases from 12.4 to 17.5, an improvement of about 41.1%. Finally, the system performs perspective transformation and tracks the detected player positions onto the top-view court image using the YOLOv5 model.

References

J. S. Sheu, C. K. Tsai, and P. T. Wang, “Driving Assistance System With Lane Change Detection,” Advances in Technology Innovation, vol. 6, no. 3, pp. 137-145, July 2021.

P. T. Wang, S. Y. Lin, and J. S. Sheu, “Vehicle Path Planning With Multicloud Computation Services,” Advances in Technology Innovation, vol. 6, no. 4, pp. 213-221, October 2021.

W. Song and S. A. Suandi, “Sign-YOLO: A Novel Lightweight Detection Model for Chinese Traffic Sign,” IEEE Access, vol. 11, pp. 113941-113951, 2023.

S. Zhang, Y. Chang, S. Wang, Y. Li, and T. Gu, “An Improved Lightweight YOLOv5 Algorithm for Detecting Railway Catenary Hanging String,” IEEE Access, vol. 11, pp. 114061-114070, 2023.

S. Han, X. Jiang, and Z. Wu, “An Improved YOLOv5 Algorithm for Wood Defect Detection Based on Attention,” IEEE Access, vol. 11, pp. 71800-71810, 2023.

Y. Guo and M. Zhang, “Blood Cell Detection Method Based on Improved YOLOv5,” IEEE Access, vol. 11, pp. 67987-67995, 2023.

Y. He, “Automatic Blood Cell Detection Based on Advanced YOLOv5s Network,” IEEE Access, vol. 12, pp. 17639-17650, 2024.

Y. Freund and R. E. Schapire, “A Decision-Theoretic Generalization of On-Line Learning and an Application to Boosting,” Journal of Computer and System Sciences, vol. 55, no. 1, pp. 119-139, August 1997.

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, Amsterdam, The Netherlands, pp. 21-37, October 2016.

N. Dalal and B. Triggs, “Histograms of Oriented Gradients for Human Detection,” IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 886-893, June 2005.

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.

A. G. Howard, M. Zhu, B. Chen, D. Kalenichenko, W. Wang, T. Weyand, et al., “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications,” https://doi.org/10.48550/arXiv.1704.04861, April 17, 2017.

M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L. C. Chen, “MobileNetV2: Inverted Residuals and Linear Bottlenecks,” IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 4510-4520, June 2018

A. Howard, M. Sandler, B. Chen, W. Wang, L. C. Chen, M. Tan, et al., “Searching for MobileNetV3,” IEEE/CVF International Conference on Computer Vision, pp. 1314-1324, October-November 2019.

J. Hu, L. Shen, and G. Sun, “Squeeze-and-Excitation Networks,” IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7132-7141, June 2018.

M. Tan and Q. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks,” Proceedings of the 36th International Conference on Machine Learning, vol. 97, pp. 6105-6114, June 2019.

M. Tan and Q. Le, “EfficientNetV2: Smaller Models and Faster Training,” Proceedings of the 38th International Conference on Machine Learning, vol. 139, pp. 10096-10106, July 2021.

T. Guo, K. Tao, Q. Hu, and Y. Shen, “Detection of Ice Hockey Players and Teams via a Two-Phase Cascaded CNN Model,” IEEE Access, vol. 8, pp. 195062-195073, 2020.

D. Acuna, “Towards Real-Time Detection and Tracking of Basketball Players Using Deep Neural Networks,” Proceedings of the 31st Conference on Neural Information Processing Systems, pp. 4-9, December 2017.

S. Hochreiter and J. Schmidhuber, “Long Short-Term Memory,” Neural Computation, vol. 9, no. 8, pp. 1735-1780, November 1997.

Z. Ivankovic, M. Rackovic, and M. Ivkovic, “Automatic Player Position Detection in Basketball Games,” Multimedia Tools and Applications, vol. 72, no. 3, pp. 2741-2767, October 2014.

P. K. Santhosh and B. Kaarthick, “An Automated Player Detection and Tracking in Basketball Game,” Computers, Materials & Continua, vol. 58, no. 3, pp. 625-639, 2019.

B. Markoski, Z. Ivanković, L. Ratgeber, P. Pecev, and D. Glušac, “Application of AdaBoost Algorithm in Basketball Player Detection,” Acta Polytechnica Hungarica, vol. 12, no. 1, pp. 189-207, 2015.