update with latest model

cl0ver012 · May 17, 2018 · f561926 · f561926
1 parent 8a35cc6
commit f561926
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo.py b/demo.py
@@ -13,7 +13,7 @@
     img_rows, img_cols = 320, 320
     channel = 4
 
-    model_weights_path = 'models/model.62-0.0524.hdf5'
+    model_weights_path = 'models/model.98-0.0459.hdf5'
     model = build_encoder_decoder()
     model.load_weights(model_weights_path)
     print(model.summary())

diff --git a/train_refinement.py b/train_refinement.py
@@ -17,7 +17,7 @@
     early_stop = EarlyStopping('val_loss', patience=patience)
     reduce_lr = ReduceLROnPlateau('val_loss', factor=0.1, patience=int(patience / 4), verbose=1)
 
-    pretrained_path = 'models/'
+    pretrained_path = 'models/model.98-0.0459.hdf5'
     encoder_decoder = build_encoder_decoder()
     encoder_decoder.load_weights(pretrained_path)
     # fix encoder-decoder part parameters and then update the refinement part.