timctho / convolutional-pose-machines-tensorflow

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Save Trained Model using Keras as "model.pb"

anujdutt9 opened this issue · comments

Hi. Thanks for an awesome project. I am trying to save the trained model, by loading the weights that you have provided and saving the model in the session using tf.contrib.keras.save_model.
This requires me to provide the "input/placeholder" which is input_placeholder and "output/placeholder" which I am unable to find. Can anyone please help me as to what would be the name of the "output/placeholder" for this model?
I am using the run_demo_hand_with_tracker.py file for this.
Thanks