Share via


快速入門:設定 Azure Kinect 主體追蹤

本快速入門將引導您完成在 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 檢視器,以檢查 Azure Kinect DK 是否已正確設定。

下載全身追蹤 SDK

  1. 選取連結以下載全身追蹤 SDK
  2. 在您的電腦上安裝全身追蹤 SDK。

驗證全身追蹤

啟動 Azure Kinect 全身追蹤檢視器,以檢查是否已正確設定全身追蹤 SDK。 此檢視器會隨 SDK msi 安裝程式一起安裝。 您可以在 [開始] 功能表或 <SDK Installation Path>\tools\k4abt_simple_3d_viewer.exe 中找到此檢視器。

如果您的 GPU 功能不夠強大,但仍想要測試結果,則可以透過下列命令,在命令列中啟動 Azure Kinect 主體追蹤檢視器<SDK Installation Path>\tools\k4abt_simple_3d_viewer.exe CPU

如果一切都已正確設定,則應該會出現一個視窗,內有 3D 點雲和已追蹤的身體。

Body Tracking 3D Viewer

指定 ONNX 執行階段執行環境

全身追蹤 SDK 僅支援 CPU、CUDA、DirectML (Windows) 和 TensorRT 執行環境,以推斷姿勢估計模型。 K4ABT_TRACKER_PROCESSING_MODE_GPU 預設為在 Linux 上執行 CUDA,以及在 Windows 上執行 DirectML。 已新增三個額外的模式來選取特定的執行環境:K4ABT_TRACKER_PROCESSING_MODE_GPU_CUDAK4ABT_TRACKER_PROCESSING_MODE_GPU_DIRECTMLK4ABT_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 會在盡可能不減少精確度的情況下提升約 2 倍的效能。 若要指定 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
CPU 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 的範例。

下一步