/Drowning_Detection

An ML Model that uses YOLOV7 & YOLOV8 to detect whether a user in a video is drowning or not.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Drowning_Detection

A Complex Drowning detection system that uses YOLOV7 & YOLOV8 to detect whether a user in a video is drowning or not.

Introduction

Architectures

  • YOLOV7
  • YOLOV8

YOLOV7 - PoseNet Detection

YOLOV8 - Drowning Estimation

Results

Testing the App

Conclusion

  • This project is still very limited to the Dataset I was provided inform of a video hence its overfitted

Further Improvements

  • Robust Dataset