From 79cecbd26e2002059ef5b2fc322b294d1e14b64f Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Tue, 29 Jan 2019 13:19:50 +0100
Subject: [PATCH] [utils] remove comments in the map_label function
---
bob/learn/pytorch/datasets/utils.py | 10 ----------
1 file changed, 10 deletions(-)
diff --git a/bob/learn/pytorch/datasets/utils.py b/bob/learn/pytorch/datasets/utils.py
index df1079e..ffc9e86 100644
--- a/bob/learn/pytorch/datasets/utils.py
+++ b/bob/learn/pytorch/datasets/utils.py
@@ -90,18 +90,8 @@ def map_labels(raw_labels, start_index=0):
for i in range(len(possible_labels)):
l = possible_labels[i]
- #labels[numpy.where(labels==l)[0][0]] = i + start_index
labels[numpy.where(labels==l)[0]] = i + start_index
- # -----
- # map back to native int, resolve the problem with dataset concatenation
- # it does: line 78 is now ok
- # for some reason, it was not working when the type of id labels were numpy.int64 ...
- #labels_int = []
- #for i in range(len(labels)):
- # labels_int.append(labels[i].item())
-
- #return labels_int
return labels
from torch.utils.data import Dataset
--
GitLab