lungs_var3_d8g_222f.py 文件源码

def load_scan(path):
    slices = [dicom.read_file(path + '/' + s) for s in os.listdir(path)]
    #slices.sort(key = lambda x: int(x.InstanceNumber))
    #slices.sort(key = lambda x: int(x.ImagePositionPatient[2]))  # from v 8 - BUGGY (should be float caused issues with segmenting and rescaling ....
    slices.sort(key = lambda x: float(x.ImagePositionPatient[2]))  # from v 8
    try:
        slice_thickness = np.abs(slices[0].ImagePositionPatient[2] - slices[1].ImagePositionPatient[2])
    except:
        slice_thickness = np.abs(slices[0].SliceLocation - slices[1].SliceLocation)

    for s in slices:
        s.SliceThickness = slice_thickness

    return slices