Make 3D Triangulate possible with pytorch_dlc branch and MA project by Zelin2001 · Pull Request #2729 · DeepLabCut/DeepLabCut

Rationality

I proceeded with my project, and then found the triangulate() function dependent on TensorFlow modules. Further debugging revealed more problems relating to the 'single' individual, who has a different bodypart dimension from common individuals.

Finally, I made minor changes to make triangulation work on my dataset. These changes might also help with other 3D MA users.

Changes

  • Compute triangulated data with pose_estimation_pytorch

  • Make correct MultiIndex with unique bodyparts

  • adapt to DataFrame.to_hdf() changes