小工具知识网

当前位置: 首页 > 行业知识 > 基于LIDC-IDRI肺结节肺癌数据集的放射组学机器学习分类良性和恶性肺癌(Python 全代码)全流程解析 (一)

行业知识

基于LIDC-IDRI肺结节肺癌数据集的放射组学机器学习分类良性和恶性肺癌(Python 全代码)全流程解析 (一)

2026-11-10 20:09:09 来源:互联网转载

基于LIDC-IDRI肺结节肺癌数据集的放射组学深度学习分类良性和恶性肺癌(Python 全代码)全流程解析 (一)

  • 1 LIDC-IDRI数据集介绍
    • 1.1数据集预处理
      • 1.1.1 图像读取-(python库pylidc安装)
      • 1.1.2图像读取-(读取CT图像的patch和分割标注)
      • 1.1.3图像归一化-(读取CT图像的patch和分割标注)
      • 1.2 图像标注信息预处理-(读取肺结节是良性还是恶性)
      • 1.2 肺结节图像切片处理处理-(切片并保存)
      • 1.3 总结

        第二部分传送门

        第三部分传送门

        1 LIDC-IDRI数据集介绍

        LIDC-IDRI数据集是用于医学影像分析的公开数据集,包含1010例低剂量CT扫描和人工标注的肺部结节信息。这些数据对于肺癌早期检测和算法开发至关重要。由美国国立卫生研究院(NIH)资助,该数据集为研究人员提供了宝贵资源,促进了医学影像处理和人工智能在肺部疾病诊断方面的进展,。

        数据集下载地址如下:*https://www.cancerimagingarchive.net/collection/lidc-idri/*该数据集大小为133GB,数据集的下载需要下载器和稳定的网络,否则可能失败。

        下载方面存在问题的私信我,我有完成数据集

        1.1数据集预处理

        数据集的预处理一共分为三个部分,图像的归一化,肺结节感兴趣区域分割,良恶性肺结节标注提取。

        LIDC-IDRI数据结构如下:

        1.1.1 图像读取-(python库pylidc安装)

        在预处理图像过程中,我们不需要自己写程序去解析所有的文件。已经有人早就写好了免费的库文件去解析数据集。即为python的pylidc库。该库的安装如下:

        打开anaconda prompt 输入

        pip install pylidc

        注意在安装完该库后,需要在系统用户文件夹下,新建pylidc.conf文件,为该库的运行指明文件存放的位置。

        [dicom]

        path = G:\dataset_zhang\LIDC-IDRI

        将以上路径修改为数据集存放的路径。

        测试该库是否安装成功代码如下:

        import pylidc as plfrom pylidc.utils import consensusimport osdataset_path = r'G:\dataset_zhang\LIDC-IDRI\'#修改路径哦dicom_name   = 'LIDC-IDRI-0001'PathDicom = os.path.join(example)  # 构建当前病例文件夹的完整路径# 查询当前病例的扫描数据,并将第一个扫描结果存储到scan变量中scan = pl.query(pl.Scan).filter(pl.Scan.patient_id == example).first()print(scan)

        1.1.2图像读取-(读取CT图像的patch和分割标注)

        我们以数据集中的’LIDC-IDRI-0001’病例数据为例,使用pylidc库中的按照病例文件夹名字的查询的方式,将该病例的所有的数据储存在scan中。而后提取所有专家对图像分割的标注坐标和肺结节的体素立方体。一个病例中含有多个肺结节默认提取第一个。使用的plt画图功能展示。代码如下

        import numpy as npimport matplotlib.pyplot as pltimport matplotlib.animation as manimfrom skimage.measure import find_contoursimport cv2import pylidc as plfrom pylidc.utils import consensusimport osdataset_path = r'G:\dataset_zhang\LIDC-IDRI\'dicom_name   = 'LIDC-IDRI-0001'print(dicom_name)  # 打印当前DICOM文件夹的名称PathDicom = os.path.join(dataset_path, dicom_name)  # 构建当前DICOM文件夹的完整路径# 查询当前病例的扫描数据,并将第一个扫描结果存储到scan变量中scan = pl.query(pl.Scan).filter(pl.Scan.patient_id == dicom_name).first()vol = scan.to_volume()# 将该扫描数据转换成数组形式的体积(volume)# 聚类注释(nodule annotations)以获取一组注释nods = scan.cluster_annotations()try:   anns = nods[0]   # 尝试获取第一个注释(annotation)   Malignancy = anns[0].Malignancy   # 获取注释中的恶性程度(Malignancy)信息except IndexError:   # 如果没有注释,或者无法获取第一个注释,则继续下一个DICOM文件夹   continue# 执行共识合并(consensus consolidation)和50%的一致性水平(agreement level)。# 我们在切片周围添加填充以提供上下文以进行查看。cmask, cbbox, masks = consensus(anns, clevel=0.5, pad=[(0,0), (7,7), (20,20)])# 提取相应的切片进行可视化image = vol[cbbox]k = int(0.5 * (cbbox[2].stop - cbbox[2].start))fig, ax = plt.subplots(1, 1, figsize=(5, 5))ax.imshow(vol[cbbox][:, :, k], cmap=plt.cm.gray, alpha=1)# 标记不同注释的边界colors = ['r', 'g', 'b', 'y']for j in range(len(masks)):   for c in find_contours(masks[j][:, :, k].astype(float), 0.5):       label = "Annotation %d" % (j+1)       plt.plot(c[:, 1], c[:, 0], colors[j], label=label)# 绘制50%共识轮廓线for c in find_contours(cmask[:, :, k].astype(float), 0.5):   plt.plot(c[:, 1], c[:, 0], '--k', label='50% Consensus')ax.axis('off')  # 关闭坐标轴ax.legend()  # 显示图例plt.tight_layout()  # 调整布局以适应图像plt.show()  # 显示图像

        结果如下:

        图片中,一共有4个专家对图像的分割的结果。和黑色的所有的专家对图像分割的平均的结果,一般情况下我们使用平均结果作为我们的实验参考。

        1.1.3图像归一化-(读取CT图像的patch和分割标注)

        图像归一化是一种将图像的像素值重新缩放到特定范围的处理方法,通常是将像素值映射到

        0,1或者−1,1之间。这有助于提高模型的稳定性和收敛速度,使得不同图像具有相似的数据分布,有利于深度学习模型的训练和性能提升。代码如下

        #归一化    def normalize_hu(image):#将输入图像的像素值(-4000 ~ 4000)归一化到0~1之间        MIN_BOUND = -1000.0        MAX_BOUND = 400.0        image = (image - MIN_BOUND) / (MAX_BOUND - MIN_BOUND)        image[image > 1] = 1.        image[image  

        1.2 图像标注信息预处理-(读取肺结节是良性还是恶性)

        这段代码根据变量Malignancy的不同取值(如"Highly Unlikely"、"Moderately Unlikely"等),为label1赋予相应的标签(1到5)。通过这个过程,将文本标签映射为数字标签,方便后续机器学习模型的处理和训练。

            if Malignancy == 'Highly Unlikely':        label1 = 1    elif Malignancy == 'Moderately Unlikely':        label1 = 2    elif Malignancy == 'Indeterminate':        label1 = 3    elif Malignancy == 'Moderately Suspicious':        label1 = 4    elif Malignancy == 'Highly Suspicious':        label1 = 5    print(label1)

        1.2 肺结节图像切片处理处理-(切片并保存)

        首先根据image和cmask创建了ArrayDicom和ArrayDicom_mask矩阵。然后通过定位x_、y_、z_的中心点,对图像进行切片。接着将切片后的图像和对应的掩膜图像分别保存为大小为50x50的JPEG文件。同时,将图像的标签label1与文件名写入label.txt文件中。最后,将计数器ii增加,并输出计数器的值。

        将处理后的图像进行切片和保存,并生成相应的标签信息。其中,ArrayDicom是图像数据矩阵,ArrayDicom_mask是对应的掩膜矩阵。通过对中心点定位,将图像在x、y、z方向上的切片进行大小为50x50的重采样和保存。同时,将每个图像切片的标签信息与文件名写入label.txt文件中,方便后续的数据处理和训练。

        结果如下:

        在label

        这个label文件记录了每个图像文件的标签信息。每一行包含两个部分,以空格分隔:

        第一部分是图像文件名,如 0.jpg, 1.jpg 等。

        第二部分是对应图像的标签,标签是一个整数,表示图像的类别或者属性。在这个文件中,标签的含义可能是:

        5: 高度可疑 (Highly Suspicious)

        3: 不确定 (Indeterminate)

        2: 中度可疑 (Moderately Suspicious)

        1.3 总结

        这样我们的就得到了我们需要训练的图像数据,它包括良性肺结节和恶性肺结节的2D图片和每一个肺结节的标签label。从三个不同方向上观察和处理图像,这在医学图像处理中是常见的。在这段代码中,每个方向的切片都被调整为大小为 (50, 50) 的图像,然后进行了插值以保持图像质量,并最终保存为 JPEG 图像文件。

        “如果您有任何医学图像处理和机器学习项目需要技术支持,请随时私信我哦。”

        有了这些文件我们就能够进行我们的放射组学特征的提取啦

        完整代码如下:

        import numpy as npimport matplotlib.pyplot as pltimport matplotlib.animation as manimfrom skimage.measure import find_contoursimport cv2import pylidc as plfrom pylidc.utils import consensusimport osdataset_path = r'G:\dataset_zhang\LIDC-IDRI\\'dicom_name   = 'LIDC-IDRI-0001'#归一化def normalize_hu(image):#将输入图像的像素值(-4000 ~ 4000)归一化到0~1之间    MIN_BOUND = -1000.0    MAX_BOUND = 400.0    image = (image - MIN_BOUND) / (MAX_BOUND - MIN_BOUND)    image[image > 1] = 1.    image[image  

        请保护原创,转载不注明出处,将追究负法律责任!!!

上一篇:C#中使用FreeSql进行数据库操作,并在WPF的MVVM模式下使用

下一篇:滴滴服务器崩溃怎么回事?