error in data.lua of MSRA dataset

mks0601 / V2V-PoseNet_RELEASE

Official Torch7 implementation of "V2V-PoseNet: Voxel-to-Voxel Prediction Network for Accurate 3D Hand and Human Pose Estimation from a Single Depth Map", CVPR 2018

https://arxiv.org/abs/1711.07399

MIT License

377 stars 69 forks source link

error in data.lua of MSRA dataset #29

Closed dragonbook closed 6 years ago

dragonbook commented 6 years ago

in src/data/MSRA/data.lua, line 136,137: "jointWorld = torch.Tensor(trainSz,jointNum,worldDim):zero() ...", I guess it should change the trainSz if it is in 'test' mode?

mks0601 commented 6 years ago

Oh, you're right. It should be 'testSz'. I'll fix it. Thanks for pointing out.