Skip to content

AI Based Smart Driver Distration Detection System

Notifications You must be signed in to change notification settings

varungupta31/AutoVision

Repository files navigation

The following Project is Divided into the following Modules:

  1. Driver Side Detection
    • Distraction Detection Using YOLOv4 Tiny
    • Drowsiness Detection Using Histrogram of Orient Gradients, Facial Landmarks and Eye Aspect Ratio
  2. Road Side Detections
    • Pedestrian Detection YOLOv4
    • Traffic Sign Detection Using YOLOv4

Distraction Detection Using YOLOv4 Tiny

The GPU Configurations for the model training were as follows:

The data of human hands, in the form of images, was acquired from the Open Images Dataset Version 6, by Google Open Source.

The performance metrics of the trained model were as follows:

Drowsiness Detection with HOG and EAR

The implementation workflow was as follows:

The Eye Aspect Ratio (EAR) was calculated as follows:

res.mp4

Pedestrian Detection

The dataset chosen for this was the PENN FUDAN Pedestrian Dataset. Pedestrians' data containing around 400 images with an average of 6 pedestrians per image was downloaded (approx. 2400 annotations). Several images (around 200) were also sourced from Google Images.

The trained model had the following metrics:

The model generalized well even in low-light conditions

Traffic Sign Detection

Indian traffic-sign datasets are limited. Available ones are private with restricted access. For this task, the GTSRB (German Traffic Sign Detection Benchmark) dataset containing 800 distinct images was used. Traffic signs were classified into 4 categories: Prohibitory, Danger, Mandatory and Other

TrafficSignal-1_speed.mp4
TS-India-2_conv_Trim.mp4

About

AI Based Smart Driver Distration Detection System

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published