scnuhealthy / Tensorflow_PersonLab

Tensorflow implementation of PersonLab (https://arxiv.org/abs/1803.08225)
83 stars 19 forks source link

Can we train on other dataset than Coco? #3

Closed aspenlin closed 5 years ago

aspenlin commented 5 years ago

Thanks!

aspenlin commented 5 years ago

Can you also add more specific information about each file and how to run them etc.. Like how to generate the images in demo_result?

aspenlin commented 5 years ago

I was able to generate the images in demo_result by simply running demo.py as you stated in README. What I meant is putting more annotations to each .py file so people can better understand your code though. Thanks!

scnuhealthy commented 5 years ago

Firstyly, most of the files will be used in training, and each file's function is as their names straightforwardly. Sorry for my busyness and not to put more code mements. If you are focus on how to generate the images, you need to read the paper to understand the whole decode process, because this part of code is quite complex. Next, if you want to train other datasets, the most simplest way is to change their annotations to coco format. Their are many guide on Internet to help you.