Driver Risk Assessment using Dashcam Footage and Machine Learning
The DriveSafe model uses transfer learning on a pre-trained ResNet50 model to capture image features, and adds an Long Short-Term Memory layer (LSTM) to capture temporal data from video input.
The model takes in dashcam-like video footage and predicts what happens throughout the video according to these 4 categories: safe
, collision
, tailgating
, weaving
.
© 2024 James Yap
Personal React Portfolio