一,国际惯例,写在前面
个人学习之作,不做商用,能力有限,有错误或者任何问题随时联系我。欢迎指正,共同进步。
二,基础步骤+遇到的问题
不做科普,需要基础的算法开发能力。
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!!