三维数据的边缘检测
sobel:
data_nii = sitk.ReadImage(img_path)
origin = data_nii.GetOrigin()
spacing = data_nii.GetSpacing()
direction = data_nii.GetDirection()
data_float_nii = sitk.Cast(data_nii, sitk.sitkFloat32)
sobel_op = sitk.SobelEdgeDetectionImageFilter()
sobel_sitk = sobel_op.Execute(data_float_nii)
sobel_sitk = sitk.Cast(sobel_sitk, sitk.sitkInt16)
sobel_sitk.SetOrigin(origin)
sobel_sitk.SetSpacing(spacing)
sobel_sitk.SetDirection(direction)
sitk.WriteImage(sobel_sitk, 'sobel_edge.nii.gz')