handpose_X 之 onnx runtime
相关项目地址:
-
1、手部关键点检测
项目地址:https://gitcode.net/EricLee/handpose_x
该项目中通过脚本 model2onnx.py ,将 .pth模型转为.onnx模型。
示例视频:
开源项目 - 手势识别 手势检测 手部21关键点检测
-
2、手部关键点检测 onnx 模型,onnx runtime C++模型推理。
项目地址:https://gitcode.net/EricLee/onnx_run
该项目对onnx模型进行基于onnx runtime框架的C++推理。
项目应用介绍
项目配置
1、作者开发环境: Windows10 (vs 2019 or vs 2022)
2、项目依赖
- opencv 3.4.15
安装方法:自行安装配置项目依赖。 - 如模型运行平台 GPU :
- onnxruntime-win-x64-gpu-1.3.0
- cuda_10.1.105_418.96_win10
- cudnn-10.1-windows10-x64-v7.6.5.32
安装方法:自行安装配置项目依赖。
- 如模型运行平台 CPU :
- Microsoft.ML.OnnxRuntime.1.10.0 & Microsoft.ML.OnnxRuntime.MKLML.1.6.0
安装方法:
进入 visual studio 编辑环境,选择:工具 -> NuGet 包管理器 -> 管理解决方案的 NuGet程序包,
然后选择“浏览”,在搜索栏分别输入 “Microsoft.ML.OnnxRuntime.1.10.0”、“Microsoft.ML.OnnxRuntime.MKLML.1.6.0”,进行安装。
项目依赖工具包
ONNX 模型
项目使用方法
1、如GPU运行自行确定软件已安装及硬件环境符合。
- 注意:另外通过 main.cpp 中定义 USE_CUDA ,确定是运行在GPU or CPU 硬件上。