Using the classifier

Now that we've enhanced the classifier to load random images, we'll start with choosing these random images with the exact size and shape of our training/testing images. We'll need to add placeholders for these user-provided images, so we'll add the following lines in the appropriate locations:

 tf_random_dataset = tf.placeholder(tf.float32, shape=(1, 
  image_size, image_size, num_channels),  
 name='RANDOM_DATA')random_prediction =  
 tf.nn.softmax(nn_model(tf_random_dataset, 
  weights, biases))

Next, we will grab the image provided by the user via the following command-line parameter and run our session on the image:

We will follow almost the exact sequence as we did earlier. Running a test file through the script using the -e switch will yield an extra output, as follows:

    The prediction is: 2 

Voila! We just classified an arbitrary image.