diff options
Diffstat (limited to 'tensorflow/examples/image_retraining/retrain.py')
-rw-r--r-- | tensorflow/examples/image_retraining/retrain.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/examples/image_retraining/retrain.py b/tensorflow/examples/image_retraining/retrain.py index e612eb7424..575dea5584 100644 --- a/tensorflow/examples/image_retraining/retrain.py +++ b/tensorflow/examples/image_retraining/retrain.py @@ -893,7 +893,8 @@ def main(_): print('=== MISCLASSIFIED TEST IMAGES ===') for i, test_filename in enumerate(test_filenames): if predictions[i] != test_ground_truth[i].argmax(): - print('%70s %s' % (test_filename, image_lists.keys()[predictions[i]])) + print('%70s %s' % (test_filename, + list(image_lists.keys())[predictions[i]])) # Write out the trained graph and labels with the weights stored as constants. output_graph_def = graph_util.convert_variables_to_constants( |