From bfc186ae967b185abca06eac72767e18314dd771 Mon Sep 17 00:00:00 2001 From: Anner Date: Thu, 20 Jul 2017 12:22:27 +0900 Subject: [PATCH] add './output/mask_rcnn' dir if missing I don't know where this folder is supposed to be created or initialized, but for me and others (https://github.com/CharlesShang/FastMaskRCNN/issues/91) it doesnt do that properly. This should add a simple check and fix for the missing directory structure --- download_and_convert_data.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/download_and_convert_data.py b/download_and_convert_data.py index 8a1b95a..a6f8ff2 100755 --- 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: