I'm quite new to Pose estimation field, I'm having a hard time to understand why did you multiply by a tensor (Torch.arange) in _generate_3d_integral_preds_tensor(heatmaps, num_joints, x_dim, y_dim, zdim) function at link after calculating the heatmaps ?
Furthermore, how can we calculate the variance of the heatmaps ?
Hi,
Wonderful work, and thanks for the repo !
I'm quite new to Pose estimation field, I'm having a hard time to understand why did you multiply by a tensor (Torch.arange) in _generate_3d_integral_preds_tensor(heatmaps, num_joints, x_dim, y_dim, zdim) function at link after calculating the heatmaps ?
Furthermore, how can we calculate the variance of the heatmaps ?
Thanks !