From 95879e4759785cb359d4e319f548d548323a625b Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 11:30:09 +0200
Subject: [PATCH] create dataset path in outdirs if not existing

---
 inference.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/inference.py b/inference.py
index 702d4ae..856e375 100644
--- a/inference.py
+++ b/inference.py
@@ -70,6 +70,9 @@ def main(args):
 
     dataset_name = Path(args.folder).name
 
+    if not (Path(OUTDIR)/dataset_name).exists():
+        Path(Path(OUTDIR)/dataset_name).mkdir(parents=True, exist_ok=True)
+
     infere_model(inference_loader, model, dataset_name)
 
     # detection_threshold = 0.8