lightweight openpose 成功跑起来了

前面跑openpose的时候安的最新cuda11.8。lightopenpose需要pytorch,发现pytorch官方还没有给11.8的对应版本,懒得改11.6就试着先用适配11.7的pytorch结果跑成功了。

先放lightweightopenpose的官方网址

Daniil-Osokin/lightweight-human-pose-estimation.pytorch: Fast and accurate human pose estimation in PyTorch. Contains implementation of "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose" paper. (github.com)

 下载一下预训练放在根目录下

 COCO - Common Objects in Context

再将要识别的视频也放在根目录下

虽然不知道上面这些warning怎么解决但是识别效果还可以

系统是win10, GPU为1660ti, 提前下好了对应版本的CUDA和cudnn。(虽然最后的目的是纯CPU计算,感觉用不到)。

requirement文件:

环境配的过程有点乱,还好最后跑起来了。

python 用的3.10

torch和vision是去pytorch官网(好像要魔法)查的对应cuda11.7的版本.

pycocotools是从anaconda下的2.0.6

Pycocotools :: Anaconda.org

openCV是从Conda Navigator 下的

最后运行demo如图修改参数

你可能感兴趣的:(python)