论文复现:<Beyond Static Features for Temporally Consistent 3D Human Pose and Shape from a Video>

一,国际惯例,写在前面

个人学习之作,不做商用,能力有限,有错误或者任何问题随时联系我。欢迎指正,共同进步。

二,基础步骤+遇到的问题

不做科普,需要基础的算法开发能力。

01.阅读论文,看看大概是干什么的,只能说马普所牛逼。

 

 (露出鸡脚.jpg)

02.从git 上拉代码

git clone https://github.com/hongsukchoi/TCMR_RELEASE

部署环境,安装配置文件,这是对环境的基本要求

python==3.7
numpy==1.17.5 
torch==1.4.0 
torchvision==0.5.0

如果只是为了跑通,而不是训练自己数据和模型的话,

你需要找到对应的模型和数据,这个很坑,

因为我服务器不能连外网,找了好久有好几个文件对不上,我都佛了。

大概文件目录如下:

关键的是data路径,记得自己C一个,而且内部文件如图所示

如果有问题的私信我。

完了之后会在你的服务器这个路径下,下载resnet50

 $HOME/.cache/torch/hub/checkpoints/resnet50-19c8e357.pth

 能上公网直接用就行,否则自己下好了放进去

然后运行demo.py文件,

注意的是,如果你不指定gpu,默认是1卡,用CPU跑会非常慢。

那么用demo.py运行会报错,如图所示。

RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead

 view要求tensor连续。。okok。

找到对应的文件,修改添加.contiguous()

    transforms_mat = transform_mat(
        rot_mats.contiguous().view(-1, 3, 3),
        rel_joints.contiguous().view(-1, 3, 1)).view(-1, joints.shape[1], 4, 4)

三,结果展示

测试

tcmr_demo_output

(话说这玩意怎么放大啊)

NICE!!

猜你喜欢

转载自blog.csdn.net/qq_33083551/article/details/125741830