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