cosmetic change (added prints to old tf classification)
@@ -175,12 +175,19 @@ def run_inference_on_image(image):
# Creates node ID --> English string lookup.
node_lookup = NodeLookup()
+ print ('')
+ print ('**********************************************************')
+ print ('Prediction:')
+ print ('')
+
top_k = predictions.argsort()[-FLAGS.num_top_predictions:][::-1]
+
for node_id in top_k:
human_string = node_lookup.id_to_string(node_id)
score = predictions[node_id]
print('%s (score = %.5f)' % (human_string, score))
+ print ('**********************************************************')
def maybe_download_and_extract():
"""Download and extract model tar file."""
