Skip to content

Commit eee5a4e

Browse files
committed
fix: make nitransforms/tests/test_nonlinear.py::test_densefield_map_vs_ants pass
1 parent 2c62fd9 commit eee5a4e

File tree

2 files changed

+187
-176
lines changed

2 files changed

+187
-176
lines changed

nitransforms/io/itk.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -348,9 +348,11 @@ def from_image(cls, imgobj):
348348
hdr.set_intent("vector")
349349

350350
field = np.squeeze(np.asanyarray(imgobj.dataobj))
351-
field[..., (0, 1)] *= 1.0
352-
field = field.transpose(2, 1, 0, 3)
353-
return imgobj.__class__(field, LPS @ imgobj.affine, hdr)
351+
affine = imgobj.affine
352+
midindex = (np.array(field.shape[:3]) - 1) * 0.5
353+
offset = (LPS @ affine - affine) @ (*midindex, 1)
354+
affine[:3, 3] += offset[:3]
355+
return imgobj.__class__(np.flip(field, axis=(0, 1)), imgobj.affine, hdr)
354356

355357
@classmethod
356358
def to_image(cls, imgobj):

0 commit comments

Comments
 (0)