return loaded image instead of bioio.bioimage (#439)

Co-authored-by: Benjamin Morris <[email protected]>

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)