From 05e4eab314d3b2ec8420795fc934b047058b7e82 Mon Sep 17 00:00:00 2001 From: Nicolas de Lima Date: Wed, 25 Sep 2024 11:00:12 -0300 Subject: [PATCH] Fixed missing test output folder Signed-off-by: Nicolas de Lima --- training_ws/eval.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/training_ws/eval.py b/training_ws/eval.py index 3e42e3a..f08aaab 100644 --- 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(