Merge pull request #114 from Anner-deJong/patch-1

add './output/mask_rcnn' dir if missing
CharlesShang · Jul 20, 2017 · bdae077 · bdae077
2 parents a70dcdb + bfc186a
commit bdae077
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/download_and_convert_data.py b/download_and_convert_data.py
@@ -4,6 +4,7 @@
 from __future__ import division
 from __future__ import print_function
 
+import os
 import tensorflow as tf
 
 from libs.datasets import download_and_convert_coco
@@ -21,6 +22,8 @@
 
 
 def main(_):
+  if not os.path.isdir('./output/mask_rcnn'):
+    os.makedirs('./output/mask_rcnn')
   if not FLAGS.dataset_name:
     raise ValueError('You must supply the dataset name with --dataset_name')
   if not FLAGS.dataset_dir: