From c3060befe9083987cc404c948174e3c0431b768d Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 13:08:55 +0200
Subject: [PATCH] fix : make_prediction
---
image_ref/main.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/image_ref/main.py b/image_ref/main.py
index 80ee5e1..b9b8611 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -164,10 +164,13 @@ def make_prediction_duo(model, data, f_name, f_name2):
confidence_pred_list[specie].append(output[:,0].data.cpu().numpy())
#Mono class output (only most postive paire)
output = torch.argmax(output[:,0])
+ label = torch.argmin(label)
+ print(output, label)
y_pred.extend(output)
label = torch.argmin(label)
+ print(output, label)
y_true.extend(label) # Save Truth
# constant for classes
--
GitLab