From ce6656f9bc03effedbae5572d5451eade61f0aed Mon Sep 17 00:00:00 2001
From: osimeoni <oriane.simeoni@valeo.com>
Date: Tue, 5 Oct 2021 09:01:18 +0000
Subject: [PATCH] Saving predictions: small fix
---
main_lost.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/main_lost.py b/main_lost.py
index 63cb433..a4eadb5 100755
--- a/main_lost.py
+++ b/main_lost.py
@@ -298,6 +298,9 @@ if __name__ == "__main__":
image = dataset.load_image(im_name)
visualize_predictions(image, pred, seed, scales, [w_featmap, h_featmap], vis_folder, im_name)
+ # Save the prediction
+ preds_dict[im_name] = pred
+
# Evaluation
if args.no_evaluation:
continue
--
GitLab