CUHK-CSE/wider_face
Updated • 5.68k • 52
How to use Shubham12864/YOLO26n-face 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("Shubham12864/YOLO26n-face", "<weights>.pt")
model = YOLO(weights)
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)Fine-tuned YOLO26n (nano) model for single-class face detection, trained on the WIDER FACE dataset.
pip install ultralytics huggingface_hub
from huggingface_hub import hf_hub_download
from ultralytics import YOLO
# Download weights from this Hugging Face repository
path = hf_hub_download(repo_id="Shubham12864/YOLO26n-face", filename="yolo26 widerdataset.pt")
# Load model
model = YOLO(path)
# Run detection on an image
results = model.predict("your_image.jpg")
# Display results
results[0].show()
face)