From e25f14649aa9183e266eba0372b98dd5260bf754 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Tue, 7 Apr 2020 13:12:52 +0200
Subject: [PATCH] [data.csvdataset] Fixes
---
bob/ip/binseg/data/csvdataset.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/bob/ip/binseg/data/csvdataset.py b/bob/ip/binseg/data/csvdataset.py
index 047bba6d..42b6febe 100644
--- a/bob/ip/binseg/data/csvdataset.py
+++ b/bob/ip/binseg/data/csvdataset.py
@@ -12,6 +12,9 @@ import torchvision.transforms.functional as VF
import bob.io.base
+import logging
+logger = logging.getLogger(__name__)
+
class CSVDataset(Dataset):
"""
@@ -154,4 +157,9 @@ class CSVDataset(Dataset):
if self.transform:
sample = self.transform(*sample)
- return [img_path] + sample
+ stem = img_path
+ if stem.startswith(self.root_path):
+ stem = os.path.relpath(stem, self.root_path)
+ elif stem.startswith(os.pathsep):
+ stem = stem[len(os.pathsep):]
+ return [stem] + sample
--
GitLab