我正在处理一个nifti图像，我将其转换为numpy数组。这是我用来获取图片边缘的代码：

import ants
img = ants.image_read("image.nii.gz")
img_numpy = img.numpy()

# For the edge detection

sx=ndimage.sobel(img_numpy,axis=0,mode='constant')
sy=ndimage.sobel(img_numpy,axis=1,mode='constant')
edge=np.hypot(sx,sy)

以及原始图像：

在检测到图片的边缘后，我想在这些边缘内创建一个遮罩。
这是带有边缘检测的图片：

这是我想要的面具（我在lifex上画了面具）：

我在opencv上看到过很多这样的操作，但遗憾的是，我没有找到一种将opencv与nifti图片结合使用的方法，所以我改用numpy。
谢谢
ps：这里是我使用的numpy数组的链接：img\u数组，这里是原始nifti图像的链接：nifti图像