png转nii

import nibabel as nib

def save_array_as_nii_volume(data, filename, reference_name=None):
    # img = sitk.GetImageFromArray(data)
    # if (reference_name is not None):
    #     img_ref = sitk.ReadImage(reference_name)
    #     img.CopyInformation(img_ref)
    # print('&&',filename)
    # sitk.WriteImage(img, filename)
    affine = np.eye(4)
    imgs_nii = nib.Nifti1Image(data, affine)
    nib.save(imgs_nii,filename)