Downloads · 30 days
21
9% of all-time downloads
farazv2/overlay-model-yolo
overlay-model-yolo is a machine learning model from farazv2. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for ultralytics. The card lists the license as agpl-3.0.
This model was trained to detect and segment overlay elements in images/videos using YOLOv8 segmentation.
Downloads · 30 days
21
9% of all-time downloads
All-time downloads
223
Public
Repo size
110 MB
Likes
0
Public
Click a slice to open those files.
.pt110 MB · 100%
From the Hugging Face model README
This model was trained to detect and segment overlay elements in images/videos using YOLOv8 segmentation.
This repository contains two primary model files:
best.pt: The model checkpoint with the best validation metrics seen so far.last.pt: The final checkpoint from the most recent training run, used for resuming.| Metric | Value |
|---|---|
| Box [email protected] | 0.9093 |
| Box [email protected]:0.95 | 0.7576 |
| Mask [email protected] | 0.6030 |
| Mask [email protected]:0.95 | 0.2714 |
pip install ultralytics
from ultralytics import YOLO
from huggingface_hub import hf_hub_download
# Download best model
model_path = hf_hub_download(
repo_id="farazv2/overlay-model-yolo",
filename="best.pt"
)
# Load model
model = YOLO(model_path)
# Run inference
results = model('image.jpg')
from ultralytics import YOLO
from huggingface_hub import hf_hub_download
# Download last model
model_path = hf_hub_download(
repo_id="farazv2/overlay-model-yolo",
filename="last.pt"
)
# Load model and resume
model = YOLO(model_path)
model.train(data='path/to/data.yaml', resume=True)
| Parameter | Value |
|---|---|
| Epochs | 10 (per run) |
| Image Size | 640 |
| Optimizer | AdamW |
| Initial Learning Rate | 0.001 |
| Batch Size | 24 |
| Mixed Precision | True |
| Patience | 20 |
This model is released under the AGPL-3.0 license, following Ultralytics YOLOv8 licensing.