Skip to content
Snippets Groups Projects
Commit a3cc8884 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

[segmentation.engine.dumper] Fix sample indexing

parent 38d54072
No related branches found
No related tags found
1 merge request!46Create common library
Hide whitespace changes
Inline Side-by-side
src/mednet/libs/segmentation/engine/dumper.py
+ 1
− 1
View file @ a3cc8884
...@@ -66,7 +66,7 @@ def run( ...@@ -66,7 +66,7 @@ def run(
directory where the HDF5 file was saved. directory where the HDF5 file was saved.
""" """
name = sample[1]["name"][0] name = sample[1]["name"][0]
target = sample[1]["target"][0] target = sample[0]["target"][0]
stem = pathlib.Path(name).with_suffix(".hdf5") stem = pathlib.Path(name).with_suffix(".hdf5")
dest = output_folder / stem dest = output_folder / stem
tqdm.tqdm.write(f"`{name}` -> `{str(dest)}`") tqdm.tqdm.write(f"`{name}` -> `{str(dest)}`")
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment