Need the code to predict an image for the model trained using flower_train_cnn.py

[frajos100]

Need the code to predict an image for the model trained using flower_train_cnn.py. Was trying to use conv_mnist_inference.py and match the parameters in the evaluation logic in flower_train_cnn.py.
BATCH_SIZE = 10
x = tf.placeholder(tf.float32, shape=[BATCH_SIZE, 5])
x_image = tf.reshape(x, [-1, IMAGE_SIZE, IMAGE_SIZE, 3])

Getting the following error
ValueError: Dimension size must be evenly divisible by 150528 but is 50 for 'Reshape' (op: 'Reshape') with input shapes: [10,5], [4] and with input tensors computed as partial shapes: input[1] = [?,224,224,3].

Also i suppose the same flower_train_cnn.py would work for training any tfrecord files we create using the images and label,txt by running the command build_image_data.py.
Was a bit confused http://yeephycho.github.io/2016/08/15/image-data-in-tensorflow/ following the link. and hence need your guidance.

Thanking you.