/ Модель

Характеристики

Лицензия
AGPL 3.0
Библиотека
ultralytics
Опубликована
6 ноя 2025

YOLO-head-detection

Поиск объектовИзображение → bounding_boxes

Набор моделей YOLOv8 и YOLO11 для обнаружения человеческих голов на изображениях, в том числе при подсчёте толпы.

811скачиваний за 30 дней1лайквеса обновлены7 окт 2026

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

wesjos/YOLO-head-detection — набор из семи контрольных точек YOLOv8 и YOLO11 для обнаружения человеческих голов на изображениях. Модели обучены на примерно 15 000 изображений для одноклассовой детекции; индекс класса 0 соответствует голове. Разработчик указывает, что данные получены с камер двух ракурсов: сверху и под углом, и предназначены, в частности, для задач подсчёта людей в толпе.

Модель подходит для поиска голов на уличных и других сценах, подсчёта толпы, размытия изображений в целях приватности, а также как предварительный этап системы распознавания лиц. Это именно детектор голов, а не лиц. Для большинства сценариев автор рекомендует 11n-head.pt; v8n-head.pt позиционируется как самый быстрый компактный вариант, v8s-head.pt — как баланс скорости и ложных срабатываний, а варианты 11m-head.pt и v8m-head.pt — для более высокого полноты обнаружения.

Модели рассчитаны на входные изображения и выдают координаты обнаруженных объектов и классы. Репозиторий содержит веса PyTorch для Ultralytics. Автор отмечает, что YOLO11 может давать низкоуверенные ложные срабатывания на изображениях без голов, а YOLOv8 в описанном тесте на таких изображениях их не давал. Лицензия репозитория — AGPL-3.0.

Сценарии применения

  • обнаружение голов на уличных камерах
  • подсчёт людей в толпе
  • поиск голов на изображениях с камер сверху
  • поиск голов на изображениях с камер под углом
  • размытие областей с головами для приватности
  • предварительный поиск людей перед распознаванием лиц

Ограничения

Это детектор голов, а не лиц; качество может зависеть от ракурса и условий съёмки. По описанию автора, варианты YOLO11 могут выдавать низкоуверенные ложные срабатывания на изображениях без голов.

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

YOLO Head Detection

Note: This is not a face detection model. Models trained on camera datasets are used for crowd counting, including two angles: one is a top-down view and the other is an oblique view camera. A collection of seven YOLO checkpoints trained to detect human heads in images — useful for crowd counting, privacy blurring, and as a first stage before face recognition.

Trained on a dataset of 15,000 images.

🏆 Recommended: 11n-head.pt

11n-head.pt is the suggested default for most users

Six heads, six boxes — 11n-head.pt at conf 0.25

Street scene, 11m-head.pt — four people are visible, and two of the heads are boxed:

street scene

Which one should you use?

  • Default / best overall → 11n-head.pt (see above).
  • Fastest on a small model → v8n-head.pt (93 FPS, 6.2 MB).
  • Best balance → v8s-head.pt (90 FPS, 22.5 MB, and zero false boxes on our negative test).
  • Highest recall → 11m-head.pt / v8m-head.pt (more detections, ~50 FPS).

Note the pattern: every YOLOv8 checkpoint we shipped stays silent on a head-free image, while the YOLO11 family tends to emit low-confidence false boxes. If false positives cost you more than missed heads, start with v8s-head.pt.

🚀 Quick start

pip install ultralytics
from ultralytics import YOLO

# pick any checkpoint from the table above
model = YOLO("v8s-head.pt")

results = model("photo.jpg", conf=0.25, imgsz=640)

for box in results[0].boxes:
    x1, y1, x2, y2 = box.xyxy[0].tolist()
    conf = float(box.conf)
    print(f"head @ ({x1:.0f},{y1:.0f})-({x2:.0f},{y2:.0f})  conf={conf:.2f}")

results[0].save("annotated.jpg")

💡 The checkpoints do not carry class names — class index 0 is the head class. To get readable labels on plotted boxes, map it yourself:

results[0].names = {0: "head"}   # visualisation only

🎓 Training

  • Dataset size: ~15,000 images
  • Task: single-class object detection (human heads)
  • Label convention: class index 0; the checkpoints include no names mapping
  • Framework: Ultralytics — YOLOv8 and YOLO11

📄 License

Released under AGPL-3.0, inherited from the Ultralytics framework.

Citation

@misc{yolo-head-detection,
  title = {YOLO Head Detection},
  note  = {Seven YOLOv8 / YOLO11 head-detection checkpoints trained on ~15k images},
  year  = {2026}
}

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · ultralytics

Нельзя запустить

Модели Ultralytics YOLO требуют отдельного рантайма.

Ревизия весов
7f2ac50

0 моделей