Merge pull request #476 from MSco/apply-transforms-compose-h5

ENH: applytransforms: dont force compose as nii.gz
ANTsX · Jun 15, 2023 · eec6f88 · eec6f88
2 parents 75912bf + ffbf0e8
commit eec6f88
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ants/registration/apply_transforms.py b/ants/registration/apply_transforms.py
@@ -146,7 +146,10 @@ def apply_transforms(fixed, moving, transformlist,
                         '-r', f,
                         '-n', interpolator]
                 args = args + mytx
-            tfn = '%scomptx.nii.gz' % compose if compose is not None else 'NA'
+            if compose:
+                tfn = '%scomptx.nii.gz' % compose if not compose.endswith('.h5') else compose
+            else:
+                tfn = 'NA'
             if compose is not None:
                 mycompo = '[%s,1]' % tfn
                 args = ['-d', fixed.dimension,