Restore 'src_affine' for repeated inverse

liamchalcroft · Apr 6, 2023 · 77c41a8 · 77c41a8
1 parent 06defb7
commit 77c41a8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/monai/transforms/spatial/array.py b/monai/transforms/spatial/array.py
@@ -230,6 +230,7 @@ def inverse(self, data: torch.Tensor) -> torch.Tensor:
         with self.trace_transform(False):
             # we can't use `self.__call__` in case a child class calls this inverse.
             out: torch.Tensor = SpatialResample.__call__(self, data, **kw_args)
+        kw_args["src_affine"] = kw_args.get("dst_affine")
         return out