Image classification using retrained VGG16 in TensorFlow