search

sydsim / personlab-tf

implementation of PersonLab(https://arxiv.org/abs/1803.08225) using TF-slim
103 stars 19 forks source link

Pretrained model? #2

Closed cemoody closed 5 years ago

cemoody commented 6 years ago

Hey! I took another spin of the system and I've been having issues getting it to train. Rather than going through those -- do you have a pretrained model that you're willing to upload? Thank you!

sydsim commented 5 years ago

Hi @cemoody ! I fixed some bugs recently, which were made up when I wrap up the code to share. I guess it will work now. I'm truly sorry for the inconvenience to you (and all people who struggled with my buggy code). If you have still problem on training the model, I hope you to share with details (IMPORTANT!!) so we can improve this project. Also, I can send you trained model backboned with MobilenetV2. Please leave your email address or send me an email if you want.

cemoody commented 5 years ago

Thank you so much @sydsim! I'm chrisemoody@gmail.com. So excited to try out the model!

noumanriazkhan commented 5 years ago

The pretrained is here https://www.dropbox.com/s/8vzpolhvh1cexj6/model.ckpt-86162.zip

cemoody commented 5 years ago

Thank you @noumanriazkhan! I'm a pytorch person myself so learning tensorflow is going slowly. Do you happen to have an example script or notebook hanging around that loads up the model in that zip and runs it on an image?

noumanriazkhan commented 5 years ago

Actually, we are on the same boat. I also want to run inference on single image but not so well versed with TF graphs. I am working on this will hopefully bring something up.

noumanriazkhan commented 5 years ago

@cemoody I have written the script for real time or single image inference and sent a pull request. Just wait for when it's accepted.

intvo commented 5 years ago

@cemoody I have written the script for real time or single image inference and sent a pull request. Just wait for when it's accepted.

I tried to run your code and I got the following error

screen shot 2018-11-11 at 5 48 44 pm
noumanriazkhan commented 5 years ago

Hi @intvo please set the batch_size=1 in config file.

sydsim commented 5 years ago

https://drive.google.com/file/d/1v_zVLxQSXI69jFIzOZoL0pRWQ1wwoz0f/view?usp=sharing this is pretrained model based on MobileNetV2.

nitba commented 5 years ago

Hi @cemoody and @noumanriazkhan Have you measure the accuracy of model on coco dataset?

Thanks,