microsoft / MeshTransformer

Research code for CVPR 2021 paper "End-to-End Human Pose and Mesh Reconstruction with Transformers"

Home Page:https://arxiv.org/abs/2012.09760

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

运行run_metro_handmesh.py训练时,出错

464hee opened this issue · comments

当 args.num_train_epochs=1时存在训练异常
下面函数
pred_camera, pred_3d_joints, pred_vertices_sub, pred_vertices = METRO_model(images, mano_model, mesh_sampler, meta_masks=meta_masks, is_train=True)

输出pred_camera是一个维度的列表,例如:[1,2,3]
但是visualize_reconstruction需要这个pred_camera参数是一个二维列表,例如:[ [1,2,3] ]

虽然可以将 args.num_train_epochs=2实现解决,但希望能更新

我想知道你是否成功下载了训练数据集‘filename.tar’,如果下载成功了,能否共享一下方法?
十分感谢!

@464hee Good catch! I will look into this problem. I thought it is related to batch size. I am not sure if the number of epochs can also cause the problem.

@PomIsBest See my reply in another thread. Thanks.

你好,我在centos上配环境,opendr怎么都安装不上,请问我只跑一跑demo可以不安装opender么