xingyizhou / pytorch-pose-hg-3d

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

The preprocess of mpii is incorrect when input_h!=input_w #88

Closed imistyrain closed 2 years ago

imistyrain commented 4 years ago

in mpii.py, the trans_input is got by

trans_input = get_affine_transform(
      c, s, r, [self.opt.input_h, self.opt.input_w])

howerever in h36m.py

trans_input = get_affine_transform(
      c, s, r, [self.opt.input_w, self.opt.input_h])

The preporcess of mpii will be wrong when input_w != input_h