From 8ef1c71d758f628f380117b9e1fc7fc87e185527 Mon Sep 17 00:00:00 2001 From: Eddie Huang Date: Fri, 1 Jan 2021 12:11:27 -0600 Subject: [PATCH] Update imagenet_to_gcs.py fixed validation files glob --- tools/datasets/imagenet_to_gcs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/datasets/imagenet_to_gcs.py b/tools/datasets/imagenet_to_gcs.py index 9bad0d02b..614bcc54c 100644 --- a/tools/datasets/imagenet_to_gcs.py +++ b/tools/datasets/imagenet_to_gcs.py @@ -348,7 +348,7 @@ def make_shuffle_idx(n): # Glob all the validation files validation_files = sorted(tf.gfile.Glob( - os.path.join(raw_data_dir, VALIDATION_DIRECTORY, '*', '.JPEG'))) + os.path.join(raw_data_dir, VALIDATION_DIRECTORY, '*.JPEG'))) # Get validation file synset labels from labels.txt validation_synsets = tf.gfile.FastGFile(