Skip to content

Commit 1baa31b

Browse files
add filename
Signed-off-by: Yiheng Wang <[email protected]>
1 parent 27d218a commit 1baa31b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

monai/transforms/io/array.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,7 @@ def __call__(self, filename: Sequence[PathLike] | PathLike, reader: ImageReader
261261
# TODO: handle multiple filenames later
262262
buffer = reader.read(filename[0])
263263
img = reader.get_data(buffer)
264+
img.meta[Key.FILENAME_OR_OBJ] = filename[0]
264265
# TODO: check ensure channel first
265266
if self.ensure_channel_first:
266267
img = EnsureChannelFirst()(img)
@@ -281,6 +282,7 @@ def __call__(self, filename: Sequence[PathLike] | PathLike, reader: ImageReader
281282
# TODO: handle multiple filenames later
282283
buffer = reader.read(filename[0])
283284
img = reader.get_data(buffer)
285+
img.meta[Key.FILENAME_OR_OBJ] = filename[0]
284286
# TODO: check ensure channel first
285287
if self.ensure_channel_first:
286288
img = EnsureChannelFirst()(img)

0 commit comments

Comments
 (0)