From d17a91bc65b3382e4e73f64074f0adf6ef572aa2 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 16:47:47 +0200
Subject: [PATCH] fix : negative pair is always negative
---
image_ref/dataset_ref.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index 676ed89..37054d7 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -133,7 +133,7 @@ class ImageFolderDuo(data.Dataset):
if i < self.positive_prop:
label_ref = target
else :
- l = np.arange(0,len(self.classes)-1,dtype=int).tolist()
+ l = np.arange(0,len(self.classes),dtype=int).tolist()
l.remove(target)
label_ref = np.random.choice(l)
else :
--
GitLab