Emotion_Detection-Using-Yolov5

Overview

Emotion detection typically categorizes emotions into basic or complex categories such as happiness, sadness, anger, fear, surprise, disgust, and sometimes more nuanced emotions like confusion, excitement, or trust. These categories can vary depending on the specific application and dataset.

📁 Dataset Used : Emotion Detection: https://universe.roboflow.com/school-ytrhi/emotion-zjj8h

The dataset consists of two classes: 0, 1, 2, 3, 4, sad

Workflow:

Data Preparation:

• Total 1313 images for training and 144 images for validation present in 26classes.
• Create a bounding boxes with the help of label-img And makesense.ai website according to YoloV5.
• Prepare folder structure that can be accept by YoloV5.

Steps to use Yolov5:

• Cloning the YoloV5 file from official repository.
• Changing the directory of yolov5
• Installing the dependencies
• Download all versions pre-trained weights.

Steps Before Training Custom Dataset:

• Go to yolov5/data/.
• Open data.yaml
• Edit the following inside it:

1. Training and Validation file path
2. Number of classes and Class names.

Training YOLOV5 Model

• Set images size 640 with batch of 8.
• Train model around 600 epochs .
• Visualise the training metrics with the help of tensorboard.

Testing Images Using Test Data

Testing Video Demo:

Emotion.detection2.mp4