search

princeton-vl / pytorch_stacked_hourglass

Pytorch implementation of the ECCV 2016 paper "Stacked Hourglass Networks for Human Pose Estimation"
BSD 3-Clause "New" or "Revised" License
465 stars 94 forks source link

How to train on custom dataset like MSCOCO? #16

Closed WinstonDeng closed 3 years ago

WinstonDeng commented 3 years ago

Hi, thanks for your great work! I'd like to train this model on COCO dataset. How to load custom dataset and build my config for training? Looking forward to your advice.

crockwell commented 3 years ago

Thanks! Sure: our multi-person pose code (https://github.com/princeton-vl/pose-ae-train) uses COCO and has a really similar structure to this code, so I'd definitely recommend starting from there. data/coco_pose/ref.py will have a lot of helpful functions I think!

WinstonDeng commented 3 years ago

@crockwell thanks for your help!

crockwell commented 3 years ago

No problem!

NiranthS commented 3 years ago

@WinstonDeng were you able to make ref.py file for the COCO dataset. If yes, can you please share it?