Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- model_name = 'mobilenet_oid'
- detection_graph = tf.Graph()
- with detection_graph.as_default():
- od_graph_def = tf.GraphDef()
- with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:
- serialized_graph = fid.read()
- od_graph_def.ParseFromString(serialized_graph)
- tf.import_graph_def(od_graph_def, name = model_name)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement