Advertisement
Guest User

Untitled

a guest
Mar 26th, 2019
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.31 KB | None | 0 0
  1. model_name = 'mobilenet_oid'
  2. detection_graph = tf.Graph()
  3. with detection_graph.as_default():
  4. od_graph_def = tf.GraphDef()
  5. with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:
  6. serialized_graph = fid.read()
  7. od_graph_def.ParseFromString(serialized_graph)
  8. tf.import_graph_def(od_graph_def, name = model_name)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement