SimpleITK 导入dicom文件生成stl文件

要使用SimpleITK库导入DICOM文件并生成STL文件，您可以按照以下步骤操作：

import SimpleITK as sitk
import vtk

def dicom_to_stl(dicom_dir, output_stl):
    # 读取DICOM序列
    reader = sitk.ImageSeriesReader()
    dicom_names = reader.GetGDCMSeriesFileNames(dicom_dir)
    reader.SetFileNames(dicom_names)
    image = reader.Execute()

    # 使用Marching Cubes算法提取表面
    surface_extractor = sitk.MarchingCubes()
    surface_extractor.SetInput(image)
    surface_extractor.SetValue(0, 1000)  # 根据实际需求设置阈值

    # 将表面转换为vtk数据类型
    surface = surface_extractor.GetOutput()
    surface_array = sitk.GetArrayFromImage(surface)
    vtk_data = vtk.vtkImageData()
    vtk_data.SetDimensions(surface_array.shape[::-1])
    vtk_data.AllocateScalars(vtk.VTK_UNSIGNED_CHAR, 1)
    vtk_data.GetPointData().SetScalars(vtk.util.numpy_support.numpy_to_vtk(surface_array.ravel(), deep=True))

    # 使用vtkSTLWriter将表面写入STL文件
    stl_writer = vtk.vtkSTLWriter()
    stl_writer.SetFileName(output_stl)
    stl_writer.SetInputData(vtk_data)
    stl_writer.Write()

dicom_dir = "path/to/dicom/files"
output_stl = "path/to/output.stl"

dicom_to_stl(dicom_dir, output_stl)

请确保您已经安装了SimpleITK和vtk库。在代码中，您需要将`dicom_dir`替换为包含DICOM文件的目录的路径，并将`output_stl`替换为要生成的STL文件的路径。

与之前的代码示例相比，这里使用了SimpleITK库而不是pydicom库来读取DICOM文件。然后，我们使用SimpleITK的Marching Cubes算法提取表面，并将表面数据转换为vtk数据类型，最后使用vtkSTLWriter将表面写入STL文件。

请注意，阈值的设置可能需要根据您的DICOM文件的实际情况进行调整。

希望对您有帮助！