search

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
6.99k stars 1.34k forks source link

heatmap prediction #141

Closed Xavier31 closed 3 years ago

Xavier31 commented 5 years ago

Does anyone know where this -0.5 come from at the end of the heatmap prediction ? (in function get_preds_fromhm)

preds.add_(-.5)

1adrianb commented 3 years ago

I am aware that this is not useful anymore, but for future reference this is due to how the key points are discretized into the heatmaps. This helps recover some acc loss due to this conversion.