Celebrity Face Detection Computer Vision Project

Puspendu AI Vision Workspace

Updated a month ago

13

views

1

download
Classes (5)
CRISTIANO-RONALDO
ELON-MUSK
KATHERINE-LANGFORD
MARIA-SARAPOVA
VIRAT-KOHLI

Metrics

Try This Model
Drop an image or
Description

Project Description: Celebrity Face Detection using YOLOv8

The goal of this project is to create a personalized face detection model for recognizing specific celebrities, including Virat Kohli, Maria Sharapova, Elon Musk, Katherine Langford, and Cristiano Ronaldo. We gathered images of these celebrities and used the Roboflow annotation tool to manually mark their faces. After annotation, the data was exported in the YOLOv8 format.

Next, we trained a custom YOLOv8 model on this dataset to identify the faces of the chosen celebrities. The project aims to accurately detect and recognize these celebrities' faces in images, utilizing the object detection capabilities of YOLOv8 to enhance face detection performance.

You can find the more details about the project on Github by clicking in this link. To view the training logs and metrics on wandb you can click here.

Use This Trained Model

Try it in your browser, or deploy via our Hosted Inference API and other deployment methods.

Supervision

Build Computer Vision Applications Faster with Supervision

Visualize and process your model results with our reusable computer vision tools.

Cite This Project

LICENSE
CC BY 4.0

If you use this dataset in a research paper, please cite it using the following BibTeX:

                        @misc{
                            celebrity-face-detection_dataset,
                            title = { Celebrity Face Detection Dataset },
                            type = { Open Source Dataset },
                            author = { Puspendu AI Vision Workspace },
                            howpublished = { \url{ https://universe.roboflow.com/puspendu-ai-vision-workspace/celebrity-face-detection } },
                            url = { https://universe.roboflow.com/puspendu-ai-vision-workspace/celebrity-face-detection },
                            journal = { Roboflow Universe },
                            publisher = { Roboflow },
                            year = { 2024 },
                            month = { oct },
                            note = { visited on 2024-11-25 },
                            }