Оценка позы транспортных средств по ключевым точкам
Обнаруживает 14 ключевых точек для оценки позы транспортных средств на изображениях.
Описание
Что умеет эта модель?
Эта модель предназначена для оценки позы транспортных средств, определяя 14 ключевых точек на каждом автомобиле. К этим точкам относятся: четыре колеса, четыре передних и задних фары, четыре угла крыши, выхлопная труба и центральная опорная точка кузова (в соответствии с канонической схемой CarFusion, Reddy et al., CVPR 2018).
Как обучена модель?
Основные веса модели () представляют собой контрольную точку Ultralytics YOLO26-pose, обученную на протяжении 100 эпох. Также подкаталоге данного репозитория опубликована базовая модель ViTPose-S. Модель была обучена на наборе данных CarFusion.
Метрики (тестовый набор, n=12761)
Модель OKS-mAP OKS-mAP@50 PCK@0.05 Параметры Примечания
YOLO26-pose (наша)
22.0%
35.0%
49.6%
~3M Ultralytics YOLO26n-pose, 100 эпох ViTPose-S (базовая) 0.1% 13.7% — 85M Top-down; 15 эпох, требуется 100+
Применение
Для использования модели скачайте веса и инициализируйте модель YOLO. Пример использования:
Базовые веса (ViTPose-S) находятся в поддиректории и загружаются через . Инструкции по инференсу для них можно найти в README на GitHub.
Порядок ключевых точек (14):
(правое переднее колесо)
(левое переднее колесо)
(правое заднее колесо)
(левое заднее колесо)
(правая передняя фара)
(левая передняя фара)
(правая задняя фара)
(левая задняя фара)
(выхлопная труба)
(правый передний угол крыши)
(левый передний угол крыши)
(правый задний угол крыши)
(левый задний угол крыши)
(центр кузова)
Ссылки
Ссылки
Предпросмотр файлов
В этой модели нет файлов для предпросмотра.
