msg4rajesh/Yolo-v8-and-hugginface

Custom Object detection using Yolo V8 and deployment on hugginface

YOLOv8 Custom Object Detection and Deployment on Hugging Face

This repository presents a custom object detection solution using YOLOv8 and UltralyticsPlus, along with deployment on Hugging Face.

Steps

1. Install the necessary tools using "Anaconda".
2. Create a virtual environment.
3. Gather the dataset for training.
4. Annotate the collected images.
5. Set up YOLOv8 (UltralyticsPlus).
6. Train the model.
7. Evaluate and test the trained model.
8. Deploy the model on Hugging Face.

Technologies

This project utilizes a combination of open source and proprietary tools to achieve its objectives:

• Git - Version control and collaboration.
• Anaconda - Environment management.
• Simple Image Download - Image collection.
• LabelImg - Dataset annotation.
• Ultralyticsplus - for yolov8 and easy deployment on huggingface
• Hugging Face - Model deployment.

Usage

For training and evaluation details, refer to the Training folder. To understand the deployment process on Hugging Face, check out the Hugging Face folder.

License

MIT License

Free Software, Hell Yeah!