🧠

Нейросети (pioneer_rknn)

Инференс моделей компьютерного зрения на NPU: детекция позы человека и ключевых точек.

YoloPose4 методов
YoloPose() class

Загружает модель YOLO Pose из реестра pioneer_rknn.

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)
YoloPose.run()

Выполняет инференс модели и возвращает найденные позы.

run(inputs: list[np.ndarray]) -> list[YoloPose.DetectBox]
YoloPose.release()

Освобождает ресурсы RKNN-модели.

release() -> None
YoloPose.DetectBox class

Обнаруженный бокс с ключевыми точками позы.

YoloPose.DetectBox