Skip to content

Instantly share code, notes, and snippets.

@singhabhinav

singhabhinav/inception_v3.py

Last active May 11, 2018 21:02
Show Gist options
  • Save singhabhinav/502840082e44be5c44256da73c5913ce to your computer and use it in GitHub Desktop.
Save singhabhinav/502840082e44be5c44256da73c5913ce to your computer and use it in GitHub Desktop.
Download ZIP
Raw
inception_v3.py
from tensorflow.contrib.slim.nets import inception
import tensorflow.contrib.slim as slim
X = tf.placeholder(tf.float32, shape=[None, 299, 299, 3])
with slim.arg_scope(inception.inception_v3_arg_scope()):
logits, end_points = inception.inception_v3(
X, num_classes=1001, is_training=False
)
predictions = end_points["Predictions"]
saver = tf.train.Saver()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment