bearpaw / pytorch-pose

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

Mscoco preprocessing flipping #41

Open achigeor opened 6 years ago

achigeor commented 6 years ago

First of all thanks for the great repo!

In pose/datasets/mscoco.py, when the flipping happens, there is pts = shufflelr(pts, width=img.size(2), dataset='mpii').

Does the flipping / mscoco training script work, or one should implement the matchedParts for mscoco?

Thanks!

vicdu commented 6 years ago

Do you plan to federate multi-datasets training hg-net?

bearpaw commented 5 years ago

@achigeor Sorry I didn't check the correctness of this part. Did you get correct results or not?