本快速入門將引導您成功在 Azure Kinect DK 上運行人體追蹤功能。
系統需求
人體追蹤 SDK 需要在主機電腦中安裝 NVIDIA GPU。 建議的身體追蹤主機電腦要求在 系統需求 頁面中詳述。
安裝軟體
安裝最新的 NVIDIA 驅動程式
下載並安裝圖形卡的最新 NVIDIA 驅動程式。 較舊的驅動程式可能與搭配身體追蹤 SDK 重新分發的 CUDA 二進位檔不相容。
適用於 Visual Studio 2015 的 Visual C++ 可轉散發套件
下載並安裝 Visual Studio 2015 的 Visual C++ 再發行套件。
設定硬體
設定 Azure Kinect DK
啟動 Azure Kinect Viewer,檢查您的 Azure Kinect DK 是否已正確設定。
下載體感追蹤 SDK
- 選取連結以 下載身體追蹤 SDK
- 在您的電腦上安裝人體追蹤SDK。
確認身體追蹤
啟動 Azure Kinect 身體追蹤查看器,以檢查身體追蹤 SDK 是否已正確設定。 查看器會隨 SDK msi 安裝程式一起安裝。 您可以在 [開始] 選單或在 <SDK Installation Path>\tools\k4abt_simple_3d_viewer.exe
找到它。
如果您沒有足夠強大的 GPU,但仍想要測試結果,您可以透過下列命令在命令行中啟動 Azure Kinect Body Tracking Viewer: <SDK Installation Path>\tools\k4abt_simple_3d_viewer.exe CPU
如果一切都已正確設定,則應該會出現具有 3D 點雲和追蹤的物體的視窗。
指定 ONNX Runtime 的執行環境
身體追蹤 SDK 支援 CPU、CUDA、DirectML(僅限 Windows)和 TensorRT 執行環境,以推理姿勢估計模型。
K4ABT_TRACKER_PROCESSING_MODE_GPU
在 Linux 上預設為執行 CUDA,在 Windows 上則預設為執行 DirectML。 新增三個額外模式來選擇特定的執行環境:K4ABT_TRACKER_PROCESSING_MODE_GPU_CUDA
、K4ABT_TRACKER_PROCESSING_MODE_GPU_DIRECTML
與 K4ABT_TRACKER_PROCESSING_MODE_GPU_TENSORRT
。
備註
ONNX 執行時期會顯示未加速之 opcode 的警告。 您可以放心地忽略這些。
ONNX 執行環境包含控制 TensorRT 模型快取的環境變數。 建議的值如下:
- ORT_TENSORRT_ENGINE_CACHE_ENABLE=1
- ORT_TENSORRT_CACHE_PATH=“pathname”
在開始人體追蹤之前,必須先建立資料夾。
這很重要
相較於其他執行環境,TensorRT 會在推斷之前預先處理模型,以產生延伸啟動時間。 引擎的快取機制會將此功能限制在第一次執行,不過,它是實驗中的功能,而且僅針對特定模型、ONNX 執行時間版本、TensorRT 版本和 GPU 模型。
TensorRT 執行環境同時支援 FP32(預設值)和 FP16。 FP16 用近兩倍的性能提升來交換些微的精確度降低。 若要指定 FP16:
- ORT_TENSORRT_FP16_ENABLE=1
ONNX 執行時環境所需的 DLL
模式 | ORT 1.10 | CUDA 11.4.3 | CUDNN 8.2.2.26 | TensorRT 8.0.3.4 |
---|---|---|---|---|
中央處理器 | msvcp140 | - | - | - |
onnxruntime | ||||
CUDA | msvcp140 | cudart64_110 | cudnn64_8 | - |
onnxruntime | cufft64_10 | cudnn_ops_infer64_8 | ||
onnxruntime_providers_cuda | cublas64_11 | cudnn_cnn_infer64_8 | ||
onnxruntime_providers_shared | cublasLt64_11 | |||
DirectML | msvcp140 | - | - | - |
onnxruntime | ||||
directml | ||||
TensorRT | msvcp140 | cudart64_110 | - | nvinfer |
onnxruntime | cufft64_10 | nvinfer_plugin | ||
onnxruntime_providers_cuda | cublas64_11 | |||
onnxruntime_providers_shared | cublasLt64_11 | |||
onnxruntime_providers_tensorrt | nvrtc64_112_0 | |||
nvrtc-builtins64_114 |
範例
您可以在這裡找到如何使用人體追蹤 SDK 的範例。