如何从 3D Numpy 数组创建 .nii (nifti) 文件

数据挖掘 火炬 麻木的 图像分割
2022-02-14 03:11:23

我有一个预测 numpy 数组。如何从阵列制作 .nii 或 .nii.gz 掩码文件?

1个回答

您可以使用nibabel

import nibabel as nb

ni_img = nib.Nifti1Image(numpy_array, affine=np.eye(4))
nib.save(ni_img, "dicom_volume_image.nii")