Merge pull request thtrieu#519 from abagshaw/master

Only check existance of summary directory if flag is set
renlicheng · Jan 10, 2018 · d9fe7fc · d9fe7fc
2 parents 178debe + abdb910
commit d9fe7fc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/darkflow/cli.py b/darkflow/cli.py
@@ -12,8 +12,12 @@ def _get_dir(dirs):
         for d in dirs:
             this = os.path.abspath(os.path.join(os.path.curdir, d))
             if not os.path.exists(this): os.makedirs(this)
-    _get_dir([FLAGS.imgdir, FLAGS.binary, FLAGS.backup, 
-             os.path.join(FLAGS.imgdir,'out'), FLAGS.summary])
+
+    requiredDirectories = [FLAGS.imgdir, FLAGS.binary, FLAGS.backup, os.path.join(FLAGS.imgdir,'out')]
+    if FLAGS.summary:
+        requiredDirectories.append(FLAGS.summary)
+
+    _get_dir(requiredDirectories)
 
     # fix FLAGS.load to appropriate type
     try: FLAGS.load = int(FLAGS.load)