From 7f1ba42c59c9cbef9fb14f90dd30e4998238f10e Mon Sep 17 00:00:00 2001 From: Benjamin Morris Date: Fri, 11 Oct 2024 11:41:09 -0700 Subject: [PATCH] return loaded image instead of bioio.bioimage --- cyto_dl/datamodules/multidim_image.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cyto_dl/datamodules/multidim_image.py b/cyto_dl/datamodules/multidim_image.py index a5599d67..14643061 100644 --- a/cyto_dl/datamodules/multidim_image.py +++ b/cyto_dl/datamodules/multidim_image.py @@ -117,6 +117,7 @@ def get_per_file_args(self, df): "original_path": row[self.img_path_column], } ) + img_data.reverse() return img_data def _metadata_to_str(self, metadata): @@ -166,7 +167,7 @@ def _transform(self, index: int): # some monai transforms return a batch. When collated, the batch dimension gets moved to the channel dimension if self.is_batch(output_img): return [{self.out_key: img} for img in output_img] - return {self.out_key: img} + return {self.out_key: output_img} def __len__(self): return len(self.img_data)