From 9829f78ce0ef6a706b9517c25f8e360cece467c3 Mon Sep 17 00:00:00 2001 From: Andrew White Date: Mon, 27 Jul 2026 07:24:52 -0500 Subject: [PATCH] fix: _get_neg_sample must return path string --- reIDfolder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/reIDfolder.py b/reIDfolder.py index b48c0f5..39ba5b9 100755 --- a/reIDfolder.py +++ b/reIDfolder.py @@ -37,7 +37,7 @@ def _get_neg_sample(self, target): neg_index = np.argwhere(self.targets != target) neg_index = neg_index.flatten() rand = random.randint(0,len(neg_index)-1) - return self.samples[neg_index[rand]] + return self.samples[neg_index[rand]][0] def __getitem__(self, index): path, target = self.samples[index]