Merge pull request #4 from dronedeploy/basedir_options

[in progress] finishing up saving to basedir

Author: nickponline

libs/inference.py

@@ -100,7 +100,9 @@ def predict(self, imagefile, predsfile, size=1200):
         mask = category2mask(prediction)
         cv2.imwrite(predsfile, mask)
 
-def run_inference(dataset, model_name='baseline_model'):
+def run_inference(dataset, model_name='baseline_model', basedir="predictions"):
+    if not os.path.isdir(basedir):
+        os.mkdir(basedir)
 
     size = 1200
     modelpath = 'models'
@@ -116,7 +118,7 @@ def run_inference(dataset, model_name='baseline_model'):
 
         imagefile = f'{dataset}/images/{scene}-ortho.tif'
         labelfile = f'{dataset}/labels/{scene}-label.png'
-        predsfile = f"{scene}-prediction.png"
+        predsfile = f"{basedir}/{scene}-prediction.png"
 
         if not os.path.exists(imagefile):
             #print(f"image {imagefile} not found, skipping.")

libs/inference_keras.py

@@ -53,7 +53,9 @@ def run_inference_on_file(imagefile, predsfile, model, size=300):
     mask = category2mask(prediction)
     Image.fromarray(mask).save(predsfile)
 
-def run_inference(dataset, model=None, model_path=None, basedir='.'):
+def run_inference(dataset, model=None, model_path=None, basedir='predictions'):
+    if not os.path.isdir(basedir):
+        os.mkdir(basedir)
     if model is None and model_path is None:
         raise Exception("model or model_path required")