Skip to content
Snippets Groups Projects
Commit d17a91bc authored by Schneider Leo's avatar Schneider Leo
Browse files

fix : negative pair is always negative

parent 9e22d1c4
No related branches found
No related tags found
No related merge requests found
...@@ -133,7 +133,7 @@ class ImageFolderDuo(data.Dataset): ...@@ -133,7 +133,7 @@ class ImageFolderDuo(data.Dataset):
if i < self.positive_prop: if i < self.positive_prop:
label_ref = target label_ref = target
else : 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) l.remove(target)
label_ref = np.random.choice(l) label_ref = np.random.choice(l)
else : else :
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment