fix : negative pair is always negative

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]