Skip to content

An end-to-end pipeline leveraging YOLOv8 to detect sponsor logos on sports jerseys, with a FastAPI backend for model inference and a Streamlit frontend for seamless image upload and detection results.

Notifications You must be signed in to change notification settings

Fatha27/Sponsor-detector-YOLOv8

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Dataset

Source:

The dataset consists of 1,425 images of Real Madrid and Barcelona players wearing jerseys, sourced from Roboflow.

Classes:

The target classes for detection are:

  • Nike
  • Adidas
  • Spotify
  • Emirates

Preprocessing:

Data augmentation techniques were applied to enhance the dataset, and the images were annotated using Roboflow's annotation tools.

Model

Architecture:

A custom YOLOv8 model was trained on the dataset to accurately detect the four target classes.

Training:

The model was trained to achieve optimal performance in recognizing the sponsor logos under various conditions.

Results:

image

  • The performance metrics can be improved by training for more epochs or on a larger dataset.

Backend

API:

A FastAPI backend was developed to handle image uploads and model inference. The backend processes the input images and returns the detection results. image

Frontend

Interface:

A Streamlit application provides a user-friendly interface for uploading images and displaying the detection results. image image

Features:

Users can upload an image, and the detected sponsor logos will be highlighted with bounding boxes and class labels.

View the Model on Roboflow

You can view the uploaded model on Roboflow for detailed inference, information and additional insights. Visit the following link to explore the dataset, annotations, and more:

About

An end-to-end pipeline leveraging YOLOv8 to detect sponsor logos on sports jerseys, with a FastAPI backend for model inference and a Streamlit frontend for seamless image upload and detection results.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published