pyskl手势/动作识别的实现与pytorch cuda环境部署保姆教程

恭喜你,找到这篇不需要翻墙就能够成功部署的方法。在国内布置这个挺麻烦的,其他帖子会出现各种问题不能完全贯通。便宜你了。。

实话5年前我用1080训练过一个基于卷积和ltsm的手势识别,实话实说感觉比现在效果好。是因为现在的注意力都在transform和diffusion了吗,没人关注这些基础功能了。。。

前置效果

项目下载

先把项目文件搞下来。项目地址

git clone https://github.com/kennymckormick/pyskl.git

下载过程很快,完成后cd pysk进入项目根目录

python部署

conda

首先当然是安装conda,这玩意可以创建相对隔离的python环境,搞python的没这个能行?anaconda和miniconda随便装哪个都行
conda官网

安装好后执行如下命令,创建python3.8版本环境

conda create -n pyskl python=3.8

一路选y完成后输入如下指令进入环境

conda activate pyskl

然后安装mim安装器,有些软件需要用这个独立安装

pip install -U openmim

Pytorch与cuda部署

pytorch是ai算法库,和tensorflow类似。cuda是计算单元驱动。这两个东西一定要版本匹配。

一般情况下我们要先看软件环境下的pytorch版本,从而推断出适配的python版本、cuda版本,再进行安装。以我们正搞的这个举例,pyskl依赖的pytorch版本是1.5.1,我们可以在网站上查询到适配的版本是10.1。把指令复制下来直接执行即可安装
查询pytorch cuda环境版本的网站

conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch

其他依赖部署

依次执行下列指令安装依赖部署

 mim install mmcv-full==1.5.0
  mim install mmdet==2.23.0
  pip install -r requirements.txt
  pip install -e .
  pip install pyskl
  pip install mediapipe

里面有个依赖的版本有问题需要手动换一下:

 pip uninstall pillow==9.3.0
 pip  install pillow==8.4.0

这样就完成pyskl的部署了

跑一下摄像头手势识别试试,别忘了装上摄像头:

python demo\demo_gesture.py

第一次运行会自动下载训练好的模型。需要小小等待一会,接着就能看到效果了

再跑一下视频动作检测,视频是demo中自带的。成功后会生成mp4文件。

python demo/demo_skeleton.py demo/ntu_sample.avi demo/demo.mp4

你不点个赞好意思么

相关推荐
古希腊掌管学习的神44 分钟前
[机器学习]XGBoost(3)——确定树的结构
人工智能·机器学习
ZHOU_WUYI1 小时前
4.metagpt中的软件公司智能体 (ProjectManager 角色)
人工智能·metagpt
靴子学长2 小时前
基于字节大模型的论文翻译(含免费源码)
人工智能·深度学习·nlp
AI_NEW_COME3 小时前
知识库管理系统可扩展性深度测评
人工智能
海棠AI实验室3 小时前
AI的进阶之路:从机器学习到深度学习的演变(一)
人工智能·深度学习·机器学习
hunteritself3 小时前
AI Weekly『12月16-22日』:OpenAI公布o3,谷歌发布首个推理模型,GitHub Copilot免费版上线!
人工智能·gpt·chatgpt·github·openai·copilot
IT古董4 小时前
【机器学习】机器学习的基本分类-强化学习-策略梯度(Policy Gradient,PG)
人工智能·机器学习·分类
centurysee4 小时前
【最佳实践】Anthropic:Agentic系统实践案例
人工智能
mahuifa4 小时前
混合开发环境---使用编程AI辅助开发Qt
人工智能·vscode·qt·qtcreator·编程ai
四口鲸鱼爱吃盐4 小时前
Pytorch | 从零构建GoogleNet对CIFAR10进行分类
人工智能·pytorch·分类