search

DrSleep / tensorflow-deeplab-resnet

DeepLab-ResNet rebuilt in TensorFlow
MIT License
1.25k stars 429 forks source link

Exporting the checkpoint file #131

Closed Zumbalamambo closed 6 years ago

Zumbalamambo commented 7 years ago

How do I export the checkpoint file to pb graph for inference?

DrSleep commented 6 years ago

no idea. please consult official tensorflow resources for that.

noumanriazkhan commented 6 years ago

You can use this script: https://github.com/KleinYuan/tf-segmentation/blob/master/tools/graph_freezer.py Change the output_node_name if needed.