Improved Automatic Drowning Detection Approach with YOLOv8
Received: 27 August 2024 | Revised: 23 September 2024 | Accepted: 27 September 2024 | Online: 2 December 2024
Corresponding author: Nouf Alharbi
Abstract
Although swimming is a popular activity that promotes relaxation and stress relief, drowning remains a serious global problem. According to the World Health Organization (WHO), drowning is the third most common cause of death. This study delves into implementing deep learning techniques for precise drowning detection. From this point of view, a drowning detection system was designed using the YOLOv8 model, which is a powerful tool for object detection and classification tasks. Using a large dataset, the YOLOv8 model was trained to recognize drowning patterns and movements and increase the likelihood of successful rescue operations by reducing response times and improving water safety. The proposed system uses deep learning techniques and YOLOv8 technology with data augmentation techniques to enhance the model's robustness to variations in lighting, pose, and background conditions. The system performance was evaluated using the Swimming and Drowning Detection dataset achieving 90.1% accuracy compared to 88.5% with YOLOv5.
Keywords:
YOLOv8, deep learning, computer vision, object detection, drowning detectionDownloads
References
A. E. Peden, J. Passmore, A. C. Queiroga, R. Sweeney, and J. Jagnoor, "Closing the gap for drowning prevention across Europe," The Lancet Public Health, vol. 7, no. 9, pp. e728–e729, Sep. 2022.
A. Rahman, A. E. Peden, L. Ashraf, D. Ryan, A.-A. Bhuiyan, and S. Beerman, "Drowning: Global Burden, Risk Factors, and Prevention Strategies," in Oxford Research Encyclopedia of Global Public Health, 2021.
N. Alharbi, "Exploring Advance Approaches for Drowning Detection: A Review," Engineering, Technology & Applied Science Research, vol. 14, no. 4, pp. 16032–16039, Aug. 2024.
M. Sohan, T. Sai Ram, and Ch. V. Rami Reddy, "A Review on YOLOv8 and Its Advancements," in Data Intelligence and Cognitive Informatics, Tirunelveli, India, 2024, pp. 529–545.
M. Shatnawi, F. Albreiki, A. Alkhoori, and M. Alhebshi, "Deep Learning and Vision-Based Early Drowning Detection," Information, vol. 14, no. 1, Jan. 2023, Art. no. 52.
Q. He, Z. Mei, H. Zhang, and X. Xu, "Automatic Real-Time Detection of Infant Drowning Using YOLOv5 and Faster R-CNN Models Based on Video Surveillance," Journal of Social Computing, vol. 4, no. 1, pp. 62–73, Mar. 2023.
J. Bai, J. Dai, Z. Wang, and S. Yang, "A detection method of the rescue targets in the marine casualty based on improved YOLOv5s," Frontiers in Neurorobotics, vol. 16, Nov. 2022.
N. Salehi, M. Keyvanara, and S. A. Monadjemmi, "An Automatic Video-based Drowning Detection System for Swimming Pools Using Active Contours," International Journal of Image, Graphics and Signal Processing, vol. 8, no. 8, pp. 1–8, Aug. 2016.
T. Liu, X. He, L. He, and F. Yuan, "A video drowning detection device based on underwater computer vision," IET Image Processing, vol. 17, no. 6, pp. 1905–1918, 2023.
V. Sathiyapriya, K. Kaviya, R. Kavitha, R. Ajitha, and S. Jothika, "Drowning Detection System," International Journal for Research Trends and Innovation, vol. 8, no. 5, pp. 747–751, 2023.
J. Terven, D. M. Córdova-Esparza, and J. A. Romero-González, "A Comprehensive Review of YOLO Architectures in Computer Vision: From YOLOv1 to YOLOv8 and YOLO-NAS," Machine Learning and Knowledge Extraction, vol. 5, no. 4, pp. 1680–1716, 2023.
W. Chen and A. Quan-Haase, "Big Data Ethics and Politics: Toward New Understandings," Social Science Computer Review, vol. 38, no. 1, pp. 3–9, Feb. 2020.
"Swimming and Drowning Detection Dataset" Roboflow. https://universe.roboflow.com/university-g3h71/swimming-and-drowning-detection.
S. Liu, L. Qi, H. Qin, J. Shi, and J. Jia, "Path Aggregation Network for Instance Segmentation," in 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, Jun. 2018, pp. 8759–8768.
Y. Cui, D. Guo, H. Yuan, H. Gu, and H. Tang, "Enhanced YOLO Network for Improving the Efficiency of Traffic Sign Detection," Applied Sciences, vol. 14, no. 2, Jan. 2024, Art. no. 555.
W. Cullerne Bown, "Sensitivity and Specificity versus Precision and Recall, and Related Dilemmas," Journal of Classification, vol. 41, no. 2, pp. 402–426, Jul. 2024.
I. P. Sary, S. Andromeda, and E. U. Armin, "Performance Comparison of YOLOv5 and YOLOv8 Architectures in Human Detection using Aerial Images," Ultima Computing : Jurnal Sistem Komputer, vol. 15, no. 1, pp. 8–13, Jun. 2023.
J. Liang, "Confusion Matrix: Machine Learning," POGIL Activity Clearinghouse, vol. 3, no. 4, Dec. 2022.
Downloads
How to Cite
License
Copyright (c) 2024 Nouf Alharbi, Layan Aljohani, Anhar Alqasir, Taghreed Alahmadi, Rehab Alhasiri, Dalia Aldajan
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.
