Skip to content

MChehab94/TensorFlow-Save-Model-Graph

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 

Repository files navigation

A simple python script that saves the graph of any tensorflow model. The script generates meta file, checkpoints, graph, frozen graph, and optimized frozen graph.
The following params are required in order to properly save the model:

  • model_directory: Directory to store the model
  • model_name: Name of your TensorFlow model
  • input_array: Array containing the names of your input TensorFlow variables
  • output_array: Array containig the names of your output TensorFlow variables
  • forzen_name: Name of the frozen pb graph (default = "frozen.pb")
  • optimimzed_name: Name of the optimized frozen pb graph (default = "optimized.pb")

For additional information, check the official documentation here

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages