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

minor...

parent 56726b4f
No related branches found
No related tags found
No related merge requests found
...@@ -342,35 +342,35 @@ class UNET: ...@@ -342,35 +342,35 @@ class UNET:
dataset = dataset.map(self.data_function_evaluate, num_parallel_calls=8) dataset = dataset.map(self.data_function_evaluate, num_parallel_calls=8)
self.eval_dataset = dataset self.eval_dataset = dataset
def setup_pipeline(self, data_nda, label_nda, perc=0.20): # def setup_pipeline(self, data_nda, label_nda, perc=0.20):
#
num_samples = data_nda.shape[0] # num_samples = data_nda.shape[0]
num_test = int(num_samples * perc) # num_test = int(num_samples * perc)
self.num_data_samples = num_samples - num_test # self.num_data_samples = num_samples - num_test
num_train = self.num_data_samples # num_train = self.num_data_samples
#
self.train_data_nda = data_nda[0:num_train] # self.train_data_nda = data_nda[0:num_train]
self.train_label_nda = label_nda[0:num_train] # self.train_label_nda = label_nda[0:num_train]
self.test_data_nda = data_nda[num_train:] # self.test_data_nda = data_nda[num_train:]
self.test_label_nda = label_nda[num_train:] # self.test_label_nda = label_nda[num_train:]
#
trn_idxs = np.arange(self.train_data_nda.shape[0]) # trn_idxs = np.arange(self.train_data_nda.shape[0])
tst_idxs = np.arange(self.test_data_nda.shape[0]) # tst_idxs = np.arange(self.test_data_nda.shape[0])
#
np.random.shuffle(tst_idxs) # np.random.shuffle(tst_idxs)
#
self.get_train_dataset(trn_idxs) # self.get_train_dataset(trn_idxs)
self.get_test_dataset(tst_idxs) # self.get_test_dataset(tst_idxs)
#
print('datetime: ', now) # print('datetime: ', now)
print('training and test data: ') # print('training and test data: ')
print('---------------------------') # print('---------------------------')
print('num train samples: ', self.num_data_samples) # print('num train samples: ', self.num_data_samples)
print('BATCH SIZE: ', BATCH_SIZE) # print('BATCH SIZE: ', BATCH_SIZE)
print('num test samples: ', tst_idxs.shape[0]) # print('num test samples: ', tst_idxs.shape[0])
print('setup_pipeline: Done') # print('setup_pipeline: Done')
def setup_pipeline_files(self, train_data_files, train_label_files, test_data_files, test_label_files): def setup_pipeline(self, train_data_files, train_label_files, test_data_files, test_label_files):
self.train_data_files = train_data_files self.train_data_files = train_data_files
self.train_label_files = train_label_files self.train_label_files = train_label_files
...@@ -898,7 +898,7 @@ class UNET: ...@@ -898,7 +898,7 @@ class UNET:
train_label_files.sort() train_label_files.sort()
valid_label_files.sort() valid_label_files.sort()
self.setup_pipeline_files(train_data_files, train_label_files, valid_data_files, valid_label_files) self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files)
self.build_model() self.build_model()
self.build_training() self.build_training()
self.build_evaluation() self.build_evaluation()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment