轻量级openpose人体姿态检测

1.去github下载轻量级openpose人体姿态检测项目:https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch(温馨提示:github网站打不开可到gitee网站去搜索下载)
轻量级openpose人体姿态检测_第1张图片
2.下载checkpoint_iter_370000.pth模型,去我的网盘下载https://pan.baidu.com/s/1Qjwq7uFDFSt60MQBwJW7Rg密码:jybl
轻量级openpose人体姿态检测_第2张图片
3.打开这个项目,放入模型,放一点视频素材测试

轻量级openpose人体姿态检测_第3张图片

4.检测运行的代码:python human-pose/demo.py --checkpoint-path checkpoint_iter_370000.pth --video 5.mp4,运行结果图片显示

python human-pose/demo.py --checkpoint-path checkpoint_iter_370000.pth --video 5.mp4

轻量级openpose人体姿态检测_第4张图片
5.运行后显示CPU占用60%,由此说明此项目是由cpu运行,现在我来调至GPU处理图片,找到处理图片的py文件,加上代码:

import torch
torch.set_num_threads(1)

轻量级openpose人体姿态检测_第5张图片
轻量级openpose人体姿态检测_第6张图片
6. 处理后显示GPU运行了,cpu内存降到了百分之20多
轻量级openpose人体姿态检测_第7张图片
7.速度快,可实时检测,同时检测准确率高,此项目堪称完美,成功了请求兄弟们一键三连。

你可能感兴趣的:(人工智能算法,python,计算机视觉,人工智能)