xingyizhou / pytorch-pose-hg-3d

PyTorch implementation for 3D human pose estimation
GNU General Public License v3.0
615 stars 141 forks source link

A request: MPII-INF-3DHP #12

Closed bloodymeli closed 6 years ago

bloodymeli commented 6 years ago

Hi Xingyi.

Can you please make available either the post-processed MPII-INF-3DHP or the script you have used in order to pre-process it?

Thanks!

xingyizhou commented 6 years ago

Hi, The scripts you want are available in the lua torch repo (https://github.com/xingyizhou/pose-hg-3d/blob/master/src/Tools/GetMPI-INF-3D.py , https://github.com/xingyizhou/pose-hg-3d/blob/master/src/util/dataset/fusion.lua#L126 , and https://github.com/xingyizhou/pose-hg-3d/blob/master/src/util/pose-hgreg-3d.lua#L203 ) . I haven't found time to convert that part to pytorch since it is not used in training. A problem of the lua code is that I was using a pre-release version of 3DHP, for that this paper published before the 3DHP paper. I haven't re-evaluated the performance on the officially released 3DHP dataset yet (and it is in a low priority comparing to other deadlines I am rushing for). You can follow my lua torch code to evaluate on the official 3DHP dataset yourself.