From 97d686036978a2d09d5c566fcb3551ce2804f798 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 5 May 2025 15:57:17 +0200
Subject: [PATCH] rm : data prediction pred
---
image_ref/main.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index ed7d6d1..2368cda 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -238,8 +238,6 @@ def make_prediction_duo(model, data, f_name, f_name2):
label = label.cuda()
output = model(imaer, imana, img_ref)
confidence = soft_max(output)
- print(label)
- print(confidence)
confidence_pred_list[specie].append(confidence[:, 0].data.cpu().numpy())
# Mono class output (only most postive paire)
output = torch.argmax(confidence[:, 0])
--
GitLab