PULSE_AI_Models / README.md
AurevinP's picture
addreadme.
5c7b65a verified
metadata
license: mit
tags:
  - yolov8
  - object-detection
  - computer-vision
  - sports-analytics
  - football
datasets:
  - roboflow

⚽ P.U.L.S.E AI Detection Models

These are the official pre-trained object detection models for P.U.L.S.E AI (Positional & Universal Live Statistical Engine).

They are based on the Ultralytics YOLOv8 architecture and fine-tuned specifically for football (soccer) analytics.

Available Models

Filename Base Model Purpose
ball_tracker_200_epoch_92L_25M.pt YOLOv8 Detects and tracks the football on the pitch.
yolov8s_player_tracker.pt YOLOv8s Detects players, goalkeepers, and referees.

Note: The team clustering model (SigLIP + UMAP + KMeans) used by P.U.L.S.E AI is trained dynamically from the input video at runtime and does not require pre-downloaded weights.

How to Use with P.U.L.S.E AI

If you are using the P.U.L.S.E AI repository, these models will be downloaded automatically when you run the pipeline setup, or you can download them manually using the built-in script:

# From the root of the PULSE AI repository:
python download_models.py

Manual Python Usage

If you wish to use these weights independently of the main P.U.L.S.E AI pipeline, you can load them directly via the ultralytics library:

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

# 1. Download the ball tracker
ball_model_path = hf_hub_download(
    repo_id="AurevinP/PULSE_AI_Models", 
    filename="ball_tracker_200_epoch_92L_25M.pt"
)

# 2. Load into YOLO
model = YOLO(ball_model_path)

# 3. Run inference
results = model("path/to/football_video.mp4")