mks0601 / TF-SimpleHumanPose

TensorFlow implementation of "Simple Baselines for Human Pose Estimation and Tracking", ECCV 2018
335 stars 90 forks source link

Name of the output node #78

Open thiranjayamunasinghe opened 2 years ago

thiranjayamunasinghe commented 2 years ago

To export the trained model, it is required to provide the name of the output node. (For line 20 of the following script which is used to freeze the model) https://gist.githubusercontent.com/morgangiraud/249505f540a5e53a48b0c1a869d370bf/raw/6cb0b4d497925517316a92f935ce5dccb6aafd17/medium-tffreeze-1.py

I also viewed the saved model graph using tensorboard but it is hard to decide what is the correct output node. Here I attached the link for the tensorboard graph. https://drive.google.com/file/d/18RmYlaIB6mnu8_f3Bvw3sVxwlu7pItX6/view?usp=sharing

If anyone has an idea about the name of the output node, please let me know