From a3cc8884399b82e388c1e80280d6779b4472fa9b Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Fri, 28 Jun 2024 12:54:10 +0200
Subject: [PATCH] [segmentation.engine.dumper] Fix sample indexing
---
src/mednet/libs/segmentation/engine/dumper.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/mednet/libs/segmentation/engine/dumper.py b/src/mednet/libs/segmentation/engine/dumper.py
index a4822e1c..742af03e 100644
--- a/src/mednet/libs/segmentation/engine/dumper.py
+++ b/src/mednet/libs/segmentation/engine/dumper.py
@@ -66,7 +66,7 @@ def run(
directory where the HDF5 file was saved.
"""
name = sample[1]["name"][0]
- target = sample[1]["target"][0]
+ target = sample[0]["target"][0]
stem = pathlib.Path(name).with_suffix(".hdf5")
dest = output_folder / stem
tqdm.tqdm.write(f"`{name}` -> `{str(dest)}`")
--
GitLab