diff --git a/bob/ip/binseg/data/csvdataset.py b/bob/ip/binseg/data/csvdataset.py
index 047bba6d3a23a244f4a6b41260209b82237e33ca..42b6febedbdf230b86105426a8b2b02cbad26c2d 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