ENH: Minor bug fixes.

ANTsX · Oct 27, 2023 · c55a9d9 · c55a9d9
1 parent 1479d18
commit c55a9d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/antspynet/utilities/lung_segmentation.py b/antspynet/utilities/lung_segmentation.py
@@ -240,7 +240,8 @@ def lung_pulmonary_artery_segmentation(ct,
     ################################
 
     if lung_mask is None:
-        lung_mask = ants.threshold_image(lung_extraction(ct, modality="ct", verbose=verbose), 1, 3, 1, 0)
+        lung_ex = lung_extraction(ct, modality="ct", verbose=verbose)
+        lung_mask = ants.threshold_image(lung_ex['segmentation_image'], 1, 3, 1, 0)
     ct_preprocessed = ants.image_clone(ct)
     ct_preprocessed = (ct_preprocessed + 800) / (500 + 800)
     ct_preprocessed[ct_preprocessed > 1.0] = 1.0