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(