Permalink
Browse files

Small nits in action_recognition_with_tf_hub.ipynb.

PiperOrigin-RevId: 195418640
  • Loading branch information...
TensorFlow Hub Authors authored and andresusanopinto committed May 4, 2018
1 parent 90a0d40 commit 60d6d7c353d3c216ec7e32d35057322b4e1b595b
Showing with 6 additions and 4 deletions.
  1. +6 −4 examples/colab/action_recognition_with_tf_hub.ipynb
@@ -8,12 +8,15 @@
"views": {},
"default_view": {},
"provenance": [],
"collapsed_sections": []
"collapsed_sections": [
"x8Q7Un821X1A"
]
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
}
},
"accelerator": "GPU"
},
"cells": [
{
@@ -359,8 +362,7 @@
"\n",
"# Create the i3d model and get the action probabilities.\n",
"with tf.Graph().as_default():\n",
" spec = hub.load_module_spec(\"https://tfhub.dev/deepmind/i3d-kinetics-400/1\")\n",
" i3d = hub.Module(spec)\n",
" i3d = hub.Module(\"https://tfhub.dev/deepmind/i3d-kinetics-400/1\")\n",
" input_placeholder = tf.placeholder(shape=(None, None, 224, 224, 3), dtype=tf.float32)\n",
" logits = i3d(input_placeholder)\n",
" probabilities = tf.nn.softmax(logits)\n",

0 comments on commit 60d6d7c

Please sign in to comment.