Commit ad56514f authored by Philip Yang's avatar Philip Yang Committed by Hongkun Yu
Browse files

Fix MobileNet example notebook issue (#4543)

Fix variable name issue in MobileNet example notebook.
Make sure all `base_name` variables are renamed to `checkpoint_name`.
parent 6a50c338
...@@ -121,8 +121,8 @@ ...@@ -121,8 +121,8 @@
"print('Downloading from ', url)\n", "print('Downloading from ', url)\n",
"!wget {url}\n", "!wget {url}\n",
"print('Unpacking')\n", "print('Unpacking')\n",
"!tar -xvf {base_name}.tgz\n", "!tar -xvf {checkpoint_name}.tgz\n",
"checkpoint = base_name + '.ckpt'\n", "checkpoint = checkpoint_name + '.ckpt'\n",
"\n", "\n",
"display.clear_output()\n", "display.clear_output()\n",
"print('Successfully downloaded checkpoint from ', url,\n", "print('Successfully downloaded checkpoint from ', url,\n",
...@@ -349,7 +349,7 @@ ...@@ -349,7 +349,7 @@
"source": [ "source": [
"import numpy as np\n", "import numpy as np\n",
"img = np.array(PIL.Image.open('panda.jpg').resize((224, 224))).astype(np.float) / 128 - 1\n", "img = np.array(PIL.Image.open('panda.jpg').resize((224, 224))).astype(np.float) / 128 - 1\n",
"gd = tf.GraphDef.FromString(open(base_name + '_frozen.pb', 'rb').read())\n", "gd = tf.GraphDef.FromString(open(checkpoint_name + '_frozen.pb', 'rb').read())\n",
"inp, predictions = tf.import_graph_def(gd, return_elements = ['input:0', 'MobilenetV2/Predictions/Reshape_1:0'])" "inp, predictions = tf.import_graph_def(gd, return_elements = ['input:0', 'MobilenetV2/Predictions/Reshape_1:0'])"
] ]
}, },
......
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