Giters

AdhamKhalifa / Musical_instruments_detection

This is the github repo for my computer vision project. I'm using YOLOv5 and PyTorch to train a model to detect a list of musical instruments in a video in order to understand the music performance scene.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Musical_instruments_detection

In this project, I'm using YOLOv5 and PyTorch to train a model to detect a list of musical instruments in a video in order to understand the music performance scene. The instruments are: acoustic guitar, xylophone, bassoon, trumpet, and tuba. A dataset of 3,000 images was used for training and 10% of it was for validation and another 10% for testing. The model achieved an accuracy of 92%.

Here are the graphs explaining the model and training:

[1] Confusion Matrix

Confusion Matrix

[2] F1 Curve

F1 Curve

[3] P Curve

P Curve

[4] PR Curve

PR Curve

[5] R Curve

R Curve

[6] Results

Results

[7] Test batches and predictions

Test batches and predictions

Test batches and predictions

Test batches and predictions

Test batches and predictions

[8] Training batches

Training batches

Training batches

Training batches

Training batches

About

This is the github repo for my computer vision project. I'm using YOLOv5 and PyTorch to train a model to detect a list of musical instruments in a video in order to understand the music performance scene.