Merge pull request #561 from ANTsX/kk_error_check

ENH: avoid kelly_kapowski returning GMP
ANTsX · Mar 13, 2024 · e78edc2 · e78edc2
2 parents 78edec9 + 8e53271
commit e78edc2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ants/segmentation/kelly_kapowski.py b/ants/segmentation/kelly_kapowski.py
@@ -64,7 +64,7 @@ def kelly_kapowski(s, g, w, its=45, r=0.025, m=1.5, gm_label=2, wm_label=3, **kw
         s = s.clone('unsigned int')
 
     d = s.dimension
-    outimg = g.clone()
+    outimg = g.clone() * 0.0
     kellargs = {'d': d,
                 's': "[{},{},{}]".format(utils.get_pointer_string(s),gm_label,wm_label),
                 'g': g,
@@ -80,6 +80,11 @@ def kelly_kapowski(s, g, w, its=45, r=0.025, m=1.5, gm_label=2, wm_label=3, **kw
 
     libfn = utils.get_lib_fn('KellyKapowski')
     libfn(processed_kellargs)
+
+    # Check thickness is not still all zeros
+    if outimg.sum() == 0.0:
+        raise RuntimeError("KellyKapowski failed to compute thickness")
+
     return outimg