From 89573b01b775c2ee0e48635cd2807713025f11fa Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 13:33:03 +0200
Subject: [PATCH] fix : make_prediction
---
image_ref/config.py | 2 +-
image_ref/main.py | 10 ++--------
2 files changed, 3 insertions(+), 9 deletions(-)
diff --git a/image_ref/config.py b/image_ref/config.py
index 3676bcdc..3f96b9eb 100644
--- a/image_ref/config.py
+++ b/image_ref/config.py
@@ -4,7 +4,7 @@ import argparse
def load_args_contrastive():
parser = argparse.ArgumentParser()
- parser.add_argument('--epoches', type=int, default=3)
+ parser.add_argument('--epoches', type=int, default=1)
parser.add_argument('--save_inter', type=int, default=50)
parser.add_argument('--eval_inter', type=int, default=1)
parser.add_argument('--noise_threshold', type=int, default=0)
diff --git a/image_ref/main.py b/image_ref/main.py
index b9b86111..3d983bc7 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -160,18 +160,12 @@ def make_prediction_duo(model, data, f_name, f_name2):
img_ref = img_ref.cuda()
label = label.cuda()
output = model(imaer,imana,img_ref)
- print(output, label)
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
+ y_pred.append(output.tolist())
+ y_true.append(label.tolist()) # Save Truth
# constant for classes
# Build confusion matrix
--
GitLab