Skip to content

Commit 7ec3631

Browse files
authored
Merge pull request #839 from ANTsX/fix_mask_dtype
BUG: Need to clone to float
2 parents a533404 + 4409fe0 commit 7ec3631

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ants/ops/mask_image.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,9 @@ def mask_image(image, mask, level=1, binarize=False):
4646
if type(leveluse) is int or type(leveluse) is float:
4747
leveluse = [level]
4848
image_out = image.clone() * 0
49+
mask_float = ants.image_clone(mask, 'float')
4950
for mylevel in leveluse:
50-
temp = ants.threshold_image(mask, mylevel, mylevel)
51+
temp = ants.threshold_image(mask_float, mylevel, mylevel)
5152
if binarize:
5253
image_out = image_out + temp
5354
else:

0 commit comments

Comments
 (0)