Skip to content

frankcgq105/Hello-World

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

Hello-World

The first repository yes test TEST AGAIN

#below code in order to get the PB file input_layer and output_layer

import tensorflow as tf

def printTensors(pb_file):

# read pb into graph_def
with tf.gfile.GFile(pb_file, "rb") as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

# import graph_def
with tf.Graph().as_default() as graph:
    tf.import_graph_def(graph_def)

# print operations
for op in graph.get_operations():
    print(op.name)

printTensors("path-to-my-pbfile.pb")

About

The first repository

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors