hongsukchoi / TCMR_RELEASE

Official Pytorch implementation of "Beyond Static Features for Temporally Consistent 3D Human Pose and Shape from a Video", CVPR 2021
MIT License
274 stars 39 forks source link

Difference number of frames between TCMR and SPIN #32

Closed linjing7 closed 1 year ago

linjing7 commented 1 year ago

Hi, thanks for your excellent work. I found that in SPIN, the total number of frames in 3DPW dataset is 35515. But in TCMR, the 3DPW frames number is 34561. Could you please tell me what causes the difference?

hongsukchoi commented 1 year ago

Hi,

I followed VIBE. I don't remeber exactly, but it has to do with this line:
indices_to_use = np.where((dataset['joints2D'][:, :, 2] > VIS_THRESH).sum(-1) > MIN_KP)[0]

linjing7 commented 1 year ago

Okay, thank you very much!