From f561926a8b421738b4bad9ba494884d97be66410 Mon Sep 17 00:00:00 2001
From: nakayamaiwao <goodcristina2023@gmail.com>
Date: Thu, 17 May 2018 10:15:30 +0800
Subject: [PATCH] update with latest model

---
 demo.py             | 2 +-
 train_refinement.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/demo.py b/demo.py
index 9396a74..4a099b6 100644
--- 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
index d4cc55f..570bcd1 100644
--- 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.