Mirrors_Framework/FSRCNN-TensorFlow
1
0
Fork 0
mirror of https://github.com/igv/FSRCNN-TensorFlow.git synced 2026-08-18 09:07:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Append model shape to the checkpoint directory name

Browse Source
This commit is contained in:
igv
2017-08-21 14:06:21 +03:00
parent 15bc798701
commit f8af70e011
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model.py
+4 -2
View File
@@ -226,7 +226,8 @@ class FSRCNN(object):
def save(self, checkpoint_dir, step):
model_name = "FSRCNN.model"
model_dir = "%s_%s" % ("fsrcnn", self.label_size)
d, s, m = self.model_params
model_dir = "%s_%s_%s-%s-%s" % ("fsrcnn", self.label_size, d, s, m)
checkpoint_dir = os.path.join(checkpoint_dir, model_dir)
if not os.path.exists(checkpoint_dir):
@@ -238,7 +239,8 @@ class FSRCNN(object):
def load(self, checkpoint_dir):
print(" [*] Reading checkpoints...")
model_dir = "%s_%s" % ("fsrcnn", self.label_size)
d, s, m = self.model_params
model_dir = "%s_%s_%s-%s-%s" % ("fsrcnn", self.label_size, d, s, m)
checkpoint_dir = os.path.join(checkpoint_dir, model_dir)
ckpt = tf.train.get_checkpoint_state(checkpoint_dir)