-
Notifications
You must be signed in to change notification settings - Fork 213
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to make ".pb" file #2
Comments
Thanks ! You can definitely use the model definitions from that repo. I used this repo: https://github.com/gliese581gg/YOLO_tensorflow What I did is I saved the graph structure using tf.train.write_graph(), then I merged the graph with the pre-trained weights using the freeze_graph utility in the TensorFlow repo. Then I stripped the unused nodes and voila, you have a .pb graph with weights that you can use in the tensorflow android demo. Cheers ! |
Thank you!! I will try |
Great work, thank you. @natanielruiz Could you provide more details? Now I am using darknet to training a detector (only one class needs to be detected). But in this step making '.bp' file ,I always get this kind of error Thank you in advance. |
Hi Xieyulai, I have never used darknet to train a network. I belive you are also talking about a protobuff ".pb" file. I don't understand where in your code you get that error, that would be helpful to know. Cheers |
@natanielruiz Did you use the tiny-yolo version? If yes, how did you remove the Dropout Nodes(since they use ops that are not available in android - "realDiv" etc) ? Can you please share your code for removing unused nodes. |
Hello!
Thank you for your nice implementation.
I would like to know how to make ".pb" file form output of darknet.
Did you use this ( https://github.com/thtrieu/darkflow ) ?
I am really glad if you help me!
The text was updated successfully, but these errors were encountered: