- Clone the project repository to your local machine:
git clone https://github.com/shirahanesuoh-mayuri/CV-3D-Project.git
- Install the project dependencies
pip install -r requirements.txt
YOLOv5
- Navigate to the yolov8 directory
cd CV-3D-Project/code/yolov5
- Train the model
python train.py
- Evaluate on the test dataset
python evaluation.py
YOLOv8
- Navigate to the yolov8 directory
cd CV-3D-Project/code/yolov8
- Train the model
python main.py
- Evaluate on the test dataset
python evaluation.py
The first two datasets are used as our training dataset, the third one is our test dataset:
-
Driver Drowsiness Dataset (DDD): https://www.kaggle.com/datasets/ismailnasri20/driver-drowsiness-dataset-ddd/data the size of this dataset is 3GB, more than 4.1k images.
-
Drowsiness Prediction Dataset: https://www.kaggle.com/datasets/rakibuleceruet/drowsiness-prediction-dataset/data the size of this dataset is 2GB, with 9120 files.
-
Test data: https://github.com/bindujiit/Driver-Drowsiness-Dataset-D3S-/blob/main/README.md[5]