setup_train.py
@@ -7,7 +7,7 @@ def main(): random.seed() data_list = [] for subdir in glob('%s/train/*' % transform_data.data_dir): for subdir in glob('%s/train/*_update' % transform_data.data_dir): for data in glob(subdir + "/*.jpg"): data_list.append(os.path.abspath(data)) random.shuffle(data_list)