import inception_v4
import tensorflow as tf
if __name__ == '__main__':
    batch_size = 5
    height, width = 299, 299
    inputs = tf.random_uniform((batch_size, height, width, 3))
    inc = inception_v4.inception_v4_base(inputs)
    sess = tf.Session()
    saver = tf.train.Saver()
    with tf.Session() as sess:
        # Restore variables from disk.
        saver.restore(sess, "inception_v4.ckpt")
        print("Model restored.")