参见:OpenPose
这是使用Pytorch框架复现OpenPose算法模型的代码,起源于本科的毕业设计,笔者通过阅读深度理解了Gitee上的这个项目openpose-self: 一个openpose的简单复现 (gitee.com),并简化了OpenPose的网络结构,在学校的超算上重新训练了网络模型,并保存了模型文件,重新训练后的模型文件的大小是原来的一半。
模型文件较大,需单独下载,并放在models/
目录下。
链接:https://pan.baidu.com/s/1-PwZt7sOMOWx9vI9SBRinA 提取码:x2s4
最后附上COCO数据集的百度云盘下载链接。