Skip to content

GitLab

Commit 8a317572 authored by James Pruegsanusak's avatar James Pruegsanusak
Browse files

Fix indentation in obj detection tutorial

parent fe2f8b01
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
object_detection/object_detection_tutorial.ipynb
View file @ 8a317572
...@@ -140,9 +140,9 @@ ...@@ -140,9 +140,9 @@
"opener.retrieve(DOWNLOAD_BASE + MODEL_FILE, MODEL_FILE)\n", "opener.retrieve(DOWNLOAD_BASE + MODEL_FILE, MODEL_FILE)\n",
"tar_file = tarfile.open(MODEL_FILE)\n", "tar_file = tarfile.open(MODEL_FILE)\n",
"for file in tar_file.getmembers():\n", "for file in tar_file.getmembers():\n",
" file_name = os.path.basename(file.name)\n", " file_name = os.path.basename(file.name)\n",
" if 'frozen_inference_graph.pb' in file_name:\n", " if 'frozen_inference_graph.pb' in file_name:\n",
" tar_file.extract(file, os.getcwd())" " tar_file.extract(file, os.getcwd())"
] ]
}, },
{ {
...@@ -162,11 +162,11 @@ ...@@ -162,11 +162,11 @@
"source": [ "source": [
"detection_graph = tf.Graph()\n", "detection_graph = tf.Graph()\n",
"with detection_graph.as_default():\n", "with detection_graph.as_default():\n",
" od_graph_def = tf.GraphDef()\n", " od_graph_def = tf.GraphDef()\n",
" with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:\n", " with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:\n",
" serialized_graph = fid.read()\n", " serialized_graph = fid.read()\n",
" od_graph_def.ParseFromString(serialized_graph)\n", " od_graph_def.ParseFromString(serialized_graph)\n",
" tf.import_graph_def(od_graph_def, name='')" " tf.import_graph_def(od_graph_def, name='')"
] ]
}, },
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment