YOLO
Collection
1 item • Updated
How to use RJTPP/tennis-ball-detection with ultralytics:
from huggingface_hub import hf_hub_download
from ultralytics import YOLO
# pick the weights file from this repo's "Files and versions" tab
weights = hf_hub_download("RJTPP/tennis-ball-detection", "<weights>.pt")
model = YOLO(weights)
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)This repository contains a YOLOv8n model fine-tuned specifically to detect tennis balls.
The goal is to accurately detect and localize tennis balls in images and video streams using the YOLOv8 object detection framework.
Install dependencies with:
pip install ultralytics opencv-python
yolo task=detect mode=predict model=tennisball.pt source=your_image.jpg
yolo task=detect mode=predict model=tennisball.pt source=0 # webcam
This project is open-source under the Apache License 2.0 License.
Base model
Ultralytics/YOLOv8