An end-to-end AI/ML project that combines Deep Learning (Object Detection) and Data Science (EDA) to simulate key components of autonomous driving systems.
This project is divided into two major parts:
- Part 1 → Vehicle Detection using Deep Learning
- Part 2 → Tesla Autopilot Incident Analysis
This project is based on a real-world autonomous driving scenario where:
- Vehicles must be detected and localized in images
- Autopilot systems must be analyzed for road safety impact
For the complete problem statement open the Problem Statement.pdf inside the Datasets & Problem Statement folder.
- Detect vehicles in images
- Localize them using bounding boxes
- Train a deep learning model for detection
- Analyze Tesla autopilot accident data
- Perform EDA on fatalities, locations, and trends
- Understand safety implications
- Datasets & Problem Statement
- Part 1
- labels.csv
- Part 2
- Tesla-Deaths.csv
- Problem Statement.pdf
- Part 1
- Notebook
- reports/figures
- Data Cleaning.ipynb
- EDA Analysis.ipynb
- reports/figures
- src
- part1DeepLearning
- init.py
- data ingestion.py
- evaluation.py
- inference.py
- model.py
- train.py
- init.py
- part1DeepLearning
- .gitignore
- LICENSE
- README.md
- outputs
- requirements.txt
-
Data Ingestion
- Reads annotations CSV
- Splits into train/validation sets
👉data_ingestion.py
-
Model Architecture
- Faster R-CNN with MobileNet backbone
- Pretrained model fine-tuned for vehicle detection
👉model.py
-
Training
- Custom PyTorch dataset
- Bounding box + label training
- GPU support
👉train.py
-
Evaluation
- IoU-based matching
- Precision & Recall calculation
👉evaluation.py
-
Inference
- Predict bounding boxes on new images
- Draw detections and save outputs
👉inference.py
- Architecture: Faster R-CNN
- Backbone: MobileNet V3
- Task: Object Detection (Bounding Boxes + Labels)
- Data Cleaning & Preprocessing
- Exploratory Data Analysis (EDA)
- Trend analysis of:
- Accidents over time
- Death distribution
- Driver vs pedestrian impact
- Autopilot-related fatalities
Data Cleaning.ipynbEDA Analysis.ipynb
- Image dataset with bounding box annotations
- Format:
- image_id
- label
- xmin
- ymin
- xmax
- ymax
- Tesla accident dataset including:
- Date, location, deaths
- Autopilot involvement
- Vehicle & collision details
- git clone https://github.com/PrakharSri18-data/Autonomous-Driving-Deep-Learning.git
- cd Autonomous-Driving-Deep-Learning
- python -m venv venv
- venv\Scripts\activate
- pip install -r requirements.txt
- python src/part1DeepLearning/data_ingestion.py
- python src/part1DeepLearning/train.py
- python src/part1DeepLearning/evaluation.py
- python src/part1DeepLearning/inference.py
- Successfully detects vehicles with bounding boxes
- Model evaluated using Precision & Recall
- Visual outputs saved in
outputsfolder
- Dataset size & preprocessing time
- Bounding box accuracy
- Class imbalance
- Real-time performance constraints
- End-to-end ML pipeline design
- Object detection using PyTorch
- Evaluation using IoU metrics
- Real-world data analysis (Tesla autopilot)
- Combining Deep Learning + Data Science in one project
- Prakhar Srivastava
- Aspiring Data Scientist & Business Analyst | Machine Learning, Deep Learning & Generative AI Enthusiast