import matplotlib.pyplot as plt
import tensorflow as tf
from PIL import Image
import numpy as np
import sys
import cv2
import scipy.ndimage
import imutils
%matplotlib inline

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data/', one_hot=True)

sess = tf.InteractiveSession()

new_saver = tf.train.import_meta_graph('./mnist_cnn.ckpt.meta')
new_saver.restore(sess, './mnist_cnn.ckpt')

X = sess.graph.get_tensor_by_name("model3/Placeholder_1:0")
logits = sess.graph.get_tensor_by_name("model3/dense_2/BiasAdd:0")
training = sess.graph.get_tensor_by_name("model3/Placeholder:0")

image_b = mnist.validation.images[np.random.randint(0, len
(mnist.validation.images))]

plt.imshow(image_b.reshape([28, 28]), cmap='Greys')

image_b = image_b.reshape([1, 784])
result = sess.run(logits, feed_dict={X:image_b, training:False})
print("MNIST predicted Number : ", sess.run(tf.argmax(result, 1)))

KeyError                                  Traceback (most recent call last)
<ipython-input-4-e2b6114e946b> in <module>()
----> 1 new_saver = tf.train.import_meta_graph('./mnist_cnn.ckpt.meta')
      2 new_saver.restore(sess, './mnist_cnn.ckpt')

/usr/local/lib/python3.5/dist-packages/tensorflow/python/training/saver.py in import_meta_graph(meta_graph_or_file, clear_devices)
 1709   else:
 1710     return _import_meta_graph_def(
 -> 1711         read_meta_graph_file(meta_graph_or_file), clear_devices)
 1712
 1713

/usr/local/lib/python3.5/dist-packages/tensorflow/python/training/saver.py in _import_meta_graph_def(meta_graph_def, clear_devices)
 1596       node.device = ""
 1597   importer.import_graph_def(
 ->1598       input_graph_def, name="", producer_op_list=producer_op_list)
 1599
 1600   # Restores all the other collections.

/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/importer.py in import_graph_def(graph_def, input_map, return_elements, name, op_dict, producer_op_list)
    256     for node in graph_def.node:
    257       # Set any default attr values that aren't present.
--> 258       op_def = op_dict[node.op]
    259       for attr_def in op_def.attr:
    260         key = attr_def.name
KeyError: 'VariableV2'

--------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-5-fff880bca43e> in <module>()
----> 1 X = sess.graph.get_tensor_by_name("model3/Placeholder_1:0")
      2 logits = sess.graph.get_tensor_by_name("model3/dense_2/BiasAdd:0")
      3 training = sess.graph.get_tensor_by_name("model3/Placeholder:0")
/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py in get_tensor_by_name(self, name)
     2605       raise TypeError("Tensor names are strings (or similar), not %s."
     2606                       % type(name).__name__)
  -> 2607     return self.as_graph_element(name, allow_tensor=True, allow_operation=False)
     2608
     2609   def _next_id(self):
     /usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py in as_graph_element(self, obj, allow_tensor, allow_operation)
    2456
    2457     with self._lock:
 -> 2458       return self._as_graph_element_locked(obj, allow_tensor, allow_operation)
    2459
    2460   def _as_graph_element_locked(self, obj, allow_tensor, allow_operation):
   /usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py in _as_graph_element_locked(self, obj, allow_tensor, allow_operation)
   2498           raise KeyError("The name %s refers to a Tensor which does not "
   2499                          "exist. The operation, %s, does not exist in the "
-> 2500                          "graph." % (repr(name), repr(op_name)))
   2501         try:
   2502           return op.outputs[out_n]
   KeyError: "The name 'model3/Placeholder_1:0' refers to a Tensor which does not exist. The operation, 'model3/Placeholder_1', does not exist in the graph."