Fix inverse_transform

clementpoiret · Sep 1, 2021 · e5390a7 · e5390a7
1 parent 547395f
commit e5390a7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/roiloc/locator.py b/roiloc/locator.py
@@ -1,6 +1,7 @@
 from typing import Optional
 
 import ants
+import numpy as np
 from ants.core import ANTsImage
 
 from .location import crop, get_coords
@@ -128,7 +129,10 @@ def inverse_transform(self, image: ANTsImage) -> ANTsImage:
         Returns:
             ANTsImage: Inverse transformed image.
         """
-        return ants.decrop_image(image, self._image)
+        empty_image = self._image.new_image_like(
+            np.zeros_like(self._image.numpy()))
+
+        return ants.decrop_image(image, empty_image)
 
 
 def test():