Merge pull request thtrieu#408 from AlexFridman/master

Make backup flag does not require for a trailing slash
omar-Fouad · Jan 9, 2018 · 718a116 · 718a116
2 parents d932e66 + 0072464
commit 718a116
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darkflow/net/help.py b/darkflow/net/help.py
@@ -20,7 +20,7 @@ def build_train_op(self):
 
 def load_from_ckpt(self):
     if self.FLAGS.load < 0: # load lastest ckpt
-        with open(self.FLAGS.backup + 'checkpoint', 'r') as f:
+        with open(os.path.join(self.FLAGS.backup, 'checkpoint'), 'r') as f:
             last = f.readlines()[-1].strip()
             load_point = last.split(' ')[1]
             load_point = load_point.split('"')[1]