thohemp / 6DRepNet360

Official Pytorch implementation of "Towards Robust and Unconstrained Full Range of Rotation Head Pose Estimation" IEEE TIP 24
MIT License
106 stars 2 forks source link

Open Source Schedule #4

Open awarebayes opened 10 months ago

awarebayes commented 10 months ago

Greetings, I was wondering when the code will become available, I am particularly interested in conversion of CMU Panoptic dataset into Euler Rotation Degrees for heads. You had mentioned in your paper that this was previously done by WHENet, but the original repo is some esoteric keras code. So I was wondering whether we can get pytorch implementation and conversion script.

Also, there is an oversight in your evaluation script, where you subtract and add 180°: https://github.com/thohemp/6DRepNet360/blob/master/sixdrepnet360/test.py#L187

You probably copy pasted your old code. This does not affect current results (tested with AFLW), but may be a hidden footgun in the future, I suggest you fix that