bearpaw / pytorch-pose

A PyTorch toolkit for 2D Human Pose Estimation.
GNU General Public License v3.0
1.11k stars 253 forks source link

Hi, there are some problems about transform.py #94

Open wqz960 opened 5 years ago

wqz960 commented 5 years ago

inp = crop(img_trans, center, scale, [256,256]) trans_pts3d = transform_preds(torch.tensor(pts3d), center, scale, [256, 256]) I want to crop all data to [256,256], and the groundtrue landmarks is transformed by your transform_preds function. But finally the landmarks did not match the image. Can you help me to correct it? THANK YOU!

YinZ-510 commented 5 years ago

@wqz960 Hi, I also want to train this network on on my own dataset, my input picuture size is 720x1280, but I don't know how to set the scale and center. Can you give me some advice?