使用NVIDIA DeepStream创建人体姿势估计应用程序

DeepStream人类姿势估计

博客地址:https://developer.nvidia.com/blog/creating-a-human-pose-estimation-application-with-deepstream-sdk/

github 地址:https://github.com/NVIDIA-AI-IOT/deepstream_pose_estimation

人体姿势估计是计算机视觉任务,它通过在视频或照片中定位人体上的某些关键点来估计人体的配置(“姿势”)。以下应用程序作为参考,以TRTPose项目为例,在DeepStream 5.0中部署自定义姿势估计模型。

有关详细的NVIDIA Developer博客,请访问此处

先决条件

你会需要

  1. DeepStreamSDK 5.0
  2. CUDA 10.2
  3. TensorRT 7.x

入门

要开始使用,请按照以下步骤操作。

  1. 在您的平台上安装DeepStream,通过运行deepstream-app验证其是否正常运行。
  2. 最好在中克隆存储库$DEEPSTREAM_DIR/sources/apps/sample_apps
  3. 下载TRTPose模型,使用此导出实用程序将其转换为ONNX ,然后在DeepStream配置文件中设置其位置。
  4. 使用$DEEPSTREAM_DIR/libs此存储库中提供的二进制文件替换OSD二进制文件(x86或Jetson)bin/。请注意,它们在各个平台之间不兼容。
  5. 编译程序
 $ cd deepstream-pose-estimation/
 $ sudo make
 $ sudo ./deepstream-pose-estimation-app <file-uri> <output-path>
  1. 最终输出以如下形式存储在“输出路径”中 Pose_Estimation.mp4

注意:如果尚未通过提供给DeepStream的ONNX模型生成.trt引擎,则将在应用程序的第一次运行时创建一个引擎。根据您使用的系统,此过程可能需要4到10分钟。

如有任何问题,请随时在DeepStreamSDK论坛上发表新帖子。

相关推荐
<p class="MsoNormal" style="line-height: 16.5pt; background: white;" align="left"><span lang="EN-US" style="font-family: 微软雅黑, sans-serif;">PyTorch</span><span style="font-family: 微软雅黑, sans-serif;">版的</span><span lang="EN-US" style="font-family: Arial, sans-serif;">YOLOv5</span><span style="font-family: 微软雅黑, sans-serif;">是高性能的实时目标检测方法。</span><span lang="EN-US" style="font-family: Arial, sans-serif;">Jetson Nano</span><span style="font-family: 微软雅黑, sans-serif;">是英伟达含有</span><span lang="EN-US" style="font-family: Arial, sans-serif;">GPU</span><span style="font-family: 微软雅黑, sans-serif;">的人工智能硬件。本课程讲述如何部署</span><span lang="EN-US" style="font-family: Arial, sans-serif;">YOLOv5</span><span style="font-family: 微软雅黑, sans-serif;">在</span><span lang="EN-US" style="font-family: Arial, sans-serif;">Jetson Nano</span><span style="font-family: 微软雅黑, sans-serif;">开发板</span><span style="font-family: 微软雅黑, sans-serif;">上。</span><span style="font-family: 微软雅黑, sans-serif; color: #000000;">部署完成后可进行图像、视频文件和摄像头视频的实时目标检测。</span><span style="color: #222226; font-family: 微软雅黑, sans-serif;">部署时使用</span><span lang="EN-US" style="color: #222226; font-family: 微软雅黑, sans-serif;">AI</span><span style="color: #222226; font-family: 微软雅黑, sans-serif;">视频处理加速引擎</span><span lang="EN-US" style="color: #222226; font-family: 微软雅黑, sans-serif;">TensorRT</span><span style="color: #222226; font-family: 微软雅黑, sans-serif;">和</span><span lang="EN-US" style="color: #222226; font-family: 微软雅黑, sans-serif;">DeepStream</span><span style="color: #222226; font-family: 微软雅黑, sans-serif;">。</span></p> <p class="MsoNormal" style="line-height: 16.5pt; background: white;" align="left"> </p> <p class="MsoNormal" style="line-height: 16.5pt; background: white;" align="left"><span style="font-family: 微软雅黑, sans-serif;">课程内容包括:原理篇(</span><span lang="EN-US" style="font-family: Arial, sans-serif;">DeepStream</span><span style="font-family: 微软雅黑, sans-serif;">介绍</span><span style="font-family: 微软雅黑, sans-serif;">、</span><span lang="EN-US" style="font-family: Arial, sans-serif;">TensorRT</span><span style="font-family: 微软雅黑, sans-serif;">介绍、</span><span style="font-family: Arial, sans-serif;"> <span lang="EN-US">tensorrtx</span></span><span style="font-family: 微软雅黑, sans-serif;">介绍)、实践篇(<span lang="EN-US">Nano</span>硬件搭建、烧录系统镜像、安装远程登录工具、安装和测试<span lang="EN-US">DeepStream</span>、安装<span lang="EN-US">yolov5</span>、生成<span lang="EN-US">yolov5s.wts</span>文件、生成<span lang="EN-US">yolov5s.engine</span>、使用<span lang="EN-US">DeepStream</span>部署<span lang="EN-US">yolov5、</span></span><span lang="EN-US" style="font-family: 微软雅黑, sans-serif;">Nano</span><span style="font-family: 微软雅黑, sans-serif;">部署测试)</span></p> <p><img src="https://img-bss.csdnimg.cn/202104061256598707.jpg" alt="课程介绍" /></p>
©️2020 CSDN 皮肤主题: 程序猿惹谁了 设计师:白松林 返回首页