Skip to content
Snippets Groups Projects
Commit b79cb57c authored by tomrink's avatar tomrink
Browse files

more fixing

parent 17acf128
Branches
No related tags found
No related merge requests found
......@@ -123,6 +123,8 @@ def run_all(directory, out_directory):
data_valid_tiles = []
data_train_tiles = []
num_train_samples = data_train.shape[0]
num_valid_samples = data_valid.shape[0]
print(' file # done: ', cnt)
print('num_train_samples, num_valid_samples: ', num_train_samples, num_valid_samples)
total_num_train_samples += num_train_samples
......@@ -152,8 +154,8 @@ def run_all(directory, out_directory):
#np.save(out_directory+'label_train_' + str(cnt), label_train)
#np.save(out_directory+'label_valid_' + str(cnt), label_valid)
num_train_samples += data_train.shape[0]
num_valid_samples += data_valid.shape[0]
num_train_samples = data_train.shape[0]
num_valid_samples = data_valid.shape[0]
print('num_train_samples, num_valid_samples: ', num_train_samples, num_valid_samples)
total_num_train_samples += num_train_samples
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment