MediaPipe使用TFLite进行推理,在移动设备上有着优秀的性能。
尽管MediaPipe在PC中使用CPU也能达到较好的效果,但是在部分Solutions中的表现不尽如人意(比如Holistic)。
目前,MediaPipe只支持Linux的GPU加速,Windows下无法使用GPU进行推理加速。
本项目使用ONNX Runtime,为MediaPipe增加了CUDA和TensorRT支持,使Windows也能进行GPU加速。
-
为了在Windows下使用GPU加速,首先需要安装英伟达相关依赖:
-
将
lib
和include
文件夹复制到mediapipe/calculators/tensor/onnxruntime
目录下 -
将CUDA、cuDNN的
bin
和TensorRT的lib
目录添加到环境变量我的运行环境:
- CUDA 11.7.1_516.94
- cuDNN 8.5.0.96_cuda11
- TensorRT 8.4.3.1
64位Windows环境下cuDNN依赖zlib,下载解压后将
dll_x64
目录下的zlibwapi.dll
复制到cuDNN
的bin
目录下
- Face Detection
- Hand Tracking
- Pose Tracking
- Holistic Tracking
- face_landmark_with_attention模型无法使用
- 编译使用CPU的样例,也依赖onnxruntime.dll(如果有人知道如何解决,欢迎PR)。
如果LOGO或开源许可侵犯了您的权益,请立即联系我,谢谢。