Я пытаюсь использовать demo.py в https://github.com/nkolot/GraphCMR. Я заинтересован в получении соединений из выведенного SMPL изображения и визуализирую его, как описано в README этого проекта: https://github.com/gulvarol/smplpytorch.
Я также разместил вопрос здесь: https://github.com/nkolot/GraphCMR/issues/36.
То, что я пробовал, не сработало.
Я изменил https://github.com/nkolot/GraphCMR/blob/4e57dca4e9da305df99383ea6312e2b3de78c321/demo.py#L118 на pred_vertices, pred_vertices_smpl, pred_camera, smpl_pose, smpl_shape = model(...), чтобы получить smpl_pose (формы torch.Size([1, 24, 3, 3])). Затем я просто сплющил его, выполнив smpl_pose.cpu().data.numpy()[:, :, :, -1].flatten('C').reshape(1, -1), и использовал результирующие параметры (1, 72) pose в качестве входных данных в переменной pose_params демки smplpytorch .
Полученная визуализация не выглядит для меня правильной. Это правильный подход? Возможно, есть более простой способ сделать то, что я делаю. https://github.com/nkolot/GraphCMR/issues/36