/AutoVision

AI Based Smart Driver Distration Detection System

Primary LanguageJupyter Notebook

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