rishizek / tensorflow-deeplab-v3-plus

DeepLabv3+ built in TensorFlow
MIT License
833 stars 307 forks source link

Missing model meta file #70

Open vliang1 opened 4 years ago

vliang1 commented 4 years ago

The pre-trained model from Dr. Sleep model.ckpt-30358 is missing meta file. And so I cannot seem to use tools to convert to PB?

If I used the export_inference_graph.py, the output saved_model cannot seem be to open by netron. If I tried to import to tensorboard, i got the below error messages.

In short, there is a way we can convert the Dr Sleep checkpoints to a PB file? thanks.

Traceback (most recent call last): File "/home/vl933734/ProjectC/Utils/import_pb_to_tensorboard.py", line 88, in app.run(main=main, argv=[sys.argv[0]] + unparsed) File "/home/vl933734/Project/TF/TF1132/venv1132/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 125, in run _sys.exit(main(argv)) File "/home/vl933734/ProjectC/Utils/import_pb_to_tensorboard.py", line 70, in main import_to_tensorboard(FLAGS.model, FLAGS.log_dir) File "/home/vl933734/ProjectC/Utils/import_pb_to_tensorboard.py", line 59, in import_to_tensorboard graph_def.ParseFromString(f.read()) google.protobuf.message.DecodeError: Error parsing message