Fixed missing test output folder

Signed-off-by: Nicolas de Lima <ndelima@ekumenlabs.com>
ekumenlabs · Sep 27, 2024 · 05e4eab · 05e4eab
1 parent aa42ae3
commit 05e4eab
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/training_ws/eval.py b/training_ws/eval.py
@@ -212,6 +212,8 @@ def decode_output(output, labels_list, threshold=0.05):
 def main():
     """Run inference on dataset and store images with bounding boxes."""
     options, args = parse_input()
+    if not os.path.exists(options.output_folder):
+        os.makedirs(options.output_folder)
     dataset = FruitDataset(options.data_dir, get_transform())
 
     validloader = torch.utils.data.DataLoader(