tensorflow / tpu

Reference models and tools for Cloud TPUs.

Home Page:https://cloud.google.com/tpu/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

imagenet_to_gcs.py

LucasPMoreira opened this issue · comments

Trying to run this script, I noted that validation data was not being converted to TFRecord files. I found in line 351 that the command

os.path.join(raw_data_dir, VALIDATION_DIRECTORY, '*', '.JPEG')

should not have the last comma and the last string should be "*.JPEG", like below. After this change, the script ran flawless.

validation_files = sorted(tf.gfile.Glob(os.path.join(raw_data_dir, VALIDATION_DIRECTORY, '*.JPEG')))