create dataset path in outdirs if not existing

2023-10-27 11:30:09 +02:00 · 2023-10-27 11:30:09 +02:00 · 95879e4759
commit 95879e4759
parent 323d79b27a
1 changed files with 3 additions and 0 deletions
--- 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