YoloPose
🧠 Нейросети (pioneer_rknn)
класс
YoloPose()
YoloPose(model_path=None, model_name=None, target=None, device_id=None, object_thresh=0.4, nms_thresh=0.5, img_width=640, img_height=640)Загружает модель YOLO Pose из реестра pioneer_rknn.
Инициализирует RKNN-модель на NPU. Модель можно указать путём (model_path) или именем из локального реестра моделей (model_name). Класс детектирует людей и 17 ключевых точек скелета.
Параметры
| Параметр | Тип | По умолчанию | Описание |
|---|---|---|---|
model_path | str | None | None | Путь к файлу модели .rknn. |
model_name | str | None | None | Имя модели в реестре, например 'yolov8n-pose'. |
target | str | None | None | Целевая платформа RKNN. |
device_id | int | None | None | Идентификатор устройства. |
object_thresh | float | 0.4 | Порог уверенности детекции. |
nms_thresh | float | 0.5 | Порог NMS. |
img_width | int | 640 | Ширина входного изображения. |
img_height | int | 640 | Высота входного изображения. |
Возвращает
YoloPose — Экземпляр модели.
Пример
human_tracking/human_tracking_simple.py
from pioneer_rknn import YoloPose
MODEL_NAME = "yolov8n-pose"
model = YoloPose(model_name=MODEL_NAME)Примечания
- Индексы ключевых точек COCO-17: 0 — нос, 5/6 — плечи, 7/8 — локти, 9/10 — запястья, 11/12 — бёдра, 13/14 — колени, 15/16 — стопы.