visualization

[newboy111]

Hello, may I ask how the data.pkl file is generated in visualization? I hope to get your help. Thank you!

[hanjq17]

Hi,

Please refer to the comments in vis.ipynb, which tell that:

"data.pkl can be collected using the evaluation script, simply by manually saving data after testing.
Important things for visualization: input pos (x_in), target pos (x_end), the prediction (x_gmn)
and the cfg (kinematics decomposition)."

You can load your model, infer the results required above, and save them into data.pkl.

Let me know if you have further questions!