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

minor...

parent 59126a2b
Branches
No related tags found
No related merge requests found
......@@ -209,11 +209,11 @@ class UNET:
n_chans = 3
if TRIPLET:
n_chans *= 3
self.X_img = tf.keras.Input(shape=(64, 64, n_chans))
self.X_img = tf.keras.Input(shape=(None, None, n_chans))
self.inputs.append(self.X_img)
#self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
self.inputs.append(tf.keras.Input(shape=(64, 64, 3)))
self.inputs.append(tf.keras.Input(shape=(None, None, 3)))
self.flight_level = 0
......@@ -999,6 +999,13 @@ class UNET:
self.build_evaluation()
self.do_training()
def run_restore_test(self, data_path='/Users/tomrink/data/salt/train/images/', label_path='/Users/tomrink/data/salt/train/masks/', ckpt_dir=None):
self.setup_salt_pipeline(data_path, label_path)
self.build_model()
self.build_training()
self.build_evaluation()
self.restore(ckpt_dir)
def run_restore(self, filename_l1b, filename_l2, ckpt_dir):
self.setup_test_pipeline(filename_l1b, filename_l2)
self.build_model()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment