Face Detection with YOLOv8 (Fine-Tuned)

This project leverages the YOLOv8 object detection architecture, fine-tuned specifically for face detection tasks.
Trained on a publicly available image dataset, the model can accurately detect and draw bounding boxes around faces in both single and multiple images. Developed using Python and the Ultralytics YOLOv8 framework, this solution provides a fast, efficient, and lightweight approach to face detection in images.

Project Overview

This project focuses on face detection using a fine-tuned YOLOv8 model. It takes images as input and accurately identifies faces by drawing bounding boxes around them. The model is trained on a publicly available dataset, making it reliable and efficient for detecting faces in both individual images and batches of images.

Key features include:

• Training (Fine-tuning) a state-of-the-art object detection model (YOLOv8) for face detection
• Support for single or multiple image inputs
• Fast and accurate bounding box predictions around faces
• Easy integration with Python scripts for inference

The project is designed for image analysis tasks that require robust face localization without real-time video processing.

High End Architecture

flowchart TD
    A[Environment Setup] --> B[Data Preparation]
    B --> C[Model Training]
    C --> D[Evaluation & Metrics Visualization]
    D --> E[Model Saving]
    E --> F[Inference & Prediction]
    F --> G[Model Validation]

Loading

Project Architecture

The Face Detection project follows a structured pipeline leveraging the YOLOv8 model and Google Colab GPU for efficient training and inference:

1. Environment Setup

   • Check GPU availability using nvidia-smi
   • Install required libraries like ultralytics for YOLOv8
   • Mount Google Drive to access dataset and save results

2. Data Preparation

   • Load and visualize sample training images using OpenCV
   • Prepare the dataset YAML file specifying training and validation image paths, number of classes, and class names

3. Model Training

   • Fine-tune the pre-trained YOLOv8n model with the prepared dataset
   • Configure training parameters such as epochs, batch size, image size, caching, and worker threads
   • Monitor training progress through logs and loss curves

4. Evaluation and Metrics Visualization

   • Load training results from CSV files
   • Plot key metrics: box loss, precision, recall, and mean Average Precision (mAP) across epochs
   • Validate model performance on the validation dataset

5. Model Saving

   • Copy the best trained weights (best.pt) from the local run directory to Google Drive for later use

6. Inference and Prediction

   • Load the saved model weights for inference
   • Predict on single or multiple validation images, drawing bounding boxes around detected faces
   • Display predictions using Matplotlib and OpenCV
   • Save predicted output images back to Google Drive

7. Model Validation

   • Run a comprehensive validation using the YOLOv8 val() method to report final metrics on the dataset

📁 Dataset Structure

/dataset
├── images
│   ├── train
│   └── val
├── labels
│   ├── train
│   └── val
└── data.yaml

Download Dataset using the link - https://drive.google.com/file/d/11a64niIzEKQy_1wmrBPY110ToKfzP1DG/view?usp=drive_link

🚀 How to Run the Project

✅ 1. Open the Colab Notebook

Open the notebook by clicking the link below or uploading it to your own Google Drive. https://github.com/dhanya807/Face_Detection/blob/main/Face_Detection.ipynb

📓 Notebook Name: Face_Detection.ipynb
📍 Make sure it's in your Google Drive if you're opening from there.

---

✅ 2. Run the Notebook Step-by-Step

• Go to Google Colab
• Click File > Open Notebook
• Choose the uploaded .ipynb file.
• Download the dataset and upload it your file.
• Change the path to your drive path
• Click Runtime > Run all to execute all cells from top to bottom, or run manually step by step.

---

✅ 3. Requirements

Colab will auto-install dependencies, but here are key packages:

• ultralytics – YOLOv8 training and inference
• opencv-python – image loading and visualization
• matplotlib – plotting metrics

These are installed in the notebook itself, so no local installation is needed.

---

✅ 4. Google Drive Setup

Make sure your dataset and model save paths point to valid folders in your Drive.
The notebook includes code to:

• Mount Google Drive
• Access dataset from: /content/drive/MyDrive/your-dataset-folder/
• Save model weights and output predictions back to Drive

---

✅ 5. Output

• 📦 Trained model saved as: best.pt
• 🖼️ Predictions saved with bounding boxes in the runs/predict/ directory
• 📊 Evaluation metrics printed and visualized at the end

Using pretrained Model

Download the pretrained model using the link and upload it to drive

https://github.com/dhanya807/Face_Detection/blob/main/face_yolov8n_best.pt

• Download and save the colab in the drive https://github.com/dhanya807/Face_Detection/blob/main/Face_Detection_modelcheck.ipynb
• Open the colab Noetebook from Drive
• Replace the trained model path to your drive link
• Replace the prediction image path to your image path in the drive
• Click Runtime > Run all to execute all cells from top to bottom, or run manually step by step.