search

mks0601 / 3DMPPE_POSENET_RELEASE

Official PyTorch implementation of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image", ICCV 2019
MIT License
817 stars 147 forks source link

loss_coord #103

Closed nikhilchh closed 2 years ago

nikhilchh commented 2 years ago

Is the loss based on "comparing 2 maps" or are you subtracting the exact coordinates ?

In my experience I have observed people using l2 loss to compare output map and gt heatmap (masked by segmentation map).

But I could not exactly understand your implemenation of first doing soft_argmax and den comparing the coords instead of maps ? isn't that too sensitive ?

mks0601 commented 2 years ago

The loss compared two coordinates. The soft-argmax extract coordinates from heatmap in a differentiable way.