Skip to content

Instantly share code, notes, and snippets.

@aakbar5
Created March 11, 2019 13:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aakbar5/e0f44df6989abfd71485d51bd12165df to your computer and use it in GitHub Desktop.
Save aakbar5/e0f44df6989abfd71485d51bd12165df to your computer and use it in GitHub Desktop.
Tensorflow: Visualize freezed model file (.pb) in Tensorboard
To visualize freezed Tensorflow model file use Tensorboard:
- Download import_pb_to_tensorboard.py (https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/import_pb_to_tensorboard.py)
- Create a directory (/path/to/tmp/folder) where intermediate files will be created to visualize model in Tensorboard.
- Execute
- python import_pb_to_tensorboard.py --model_dir="/path/to/folder/tensorflow_model_frozen.pb" --log_dir="/path/to/tmp/folder"
NOTE: Wrapped values to param # model_dir and param # log_dir in "". Otherwise it might not work.
- tensorboard --logdir="/path/to/tmp/folder"
- By default Tensorboard runs on 6006, so open http://localhost:6006 in your web-browser to see Tensorboard.
- JFI: To inspect model without loading in web browser:
- tensorboard --inspect --logdir="/path/to/tmp/folder"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment