天天看点

轻量级openpose人体姿态检测

1.去github下载轻量级openpose人体姿态检测项目:https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch(温馨提示:github网站打不开可到gitee网站去搜索下载)

轻量级openpose人体姿态检测

2.下载checkpoint_iter_370000.pth模型,去我的网盘下载https://pan.baidu.com/s/1Qjwq7uFDFSt60MQBwJW7Rg密码:jybl

轻量级openpose人体姿态检测

3.打开这个项目,放入模型,放一点视频素材测试

轻量级openpose人体姿态检测

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人体姿态检测

5.运行后显示CPU占用60%,由此说明此项目是由cpu运行,现在我来调至GPU处理图片,找到处理图片的py文件,加上代码:

import torch
torch.set_num_threads(1)
           
轻量级openpose人体姿态检测
轻量级openpose人体姿态检测

6. 处理后显示GPU运行了,cpu内存降到了百分之20多

轻量级openpose人体姿态检测

7.速度快,可实时检测,同时检测准确率高,此项目堪称完美,成功了请求兄弟们一键三连。

继续阅读