Skip to content

Commit

Permalink
update utils.py
Browse files Browse the repository at this point in the history
  • Loading branch information
benmalef committed Oct 21, 2024
1 parent 7701a46 commit 24cf84b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion GANDLF/data/inference_dataloader_histopath.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ def __getitem__(self, idx):

# this is to ensure that channels come at the beginning
# patch = patch.transpose([2, 0, 1])
patch = patch.transpose(0) # check the documentation
patch = patch.transpose(5) # check the documentation
# this is to ensure that we always have a z-stack before applying any torchio transforms
patch = np.expand_dims(patch, axis=-1)
if self.transform is not None:
Expand Down

0 comments on commit 24cf84b

Please sign in to comment.