使用Yolov7进行LUNA16的肺结节检测

2024-03-22 21:59

本文主要是介绍使用Yolov7进行LUNA16的肺结节检测,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

1、图像预处理

1.1 从mhd文件中获取CT图像,保存3张切片


def normalizePlanes(npzarray):maxHU = 400minHU = -1000npzarray = (npzarray - minHU)/(maxHU - minHU)npzarray[npzarray>1] = 1npzarray[npzarray<0] = 0npzarray *= 255return (npzarray.astype(int))def make_mask(center,diam,z,width,height,spacing,origin):''''''mask = np.zeros([height,width])  #匹配图像#定义结节所在的体素范围v_center = (center-origin)/spacingv_diam = int(diam/spacing[0]+5)v_xmin = np.max([0,int(v_center[0]-v_diam)-5])v_xmax = np.min([width-1,int(v_center[0]+v_diam)+5])v_ymin = np.max([0,int(v_center[1]-v_diam)-5])v_ymax = np.min([height-1,int(v_center[1]+v_diam)+5])v_xrange = range(v_xmin,v_xmax+1)v_yrange = range(v_ymin,v_ymax+1)x_data = [x*spacing[0]+origin[0] for x in range(width)]y_data = [x*spacing[1]+origin[1] for x in range(height)]#结节周围全都填充1for v_x in v_xrange:for v_y in v_yrange:p_x = spacing[0]*v_x + origin[0]p_y = spacing[1]*v_y + origin[1]if np.linalg.norm(center-np.array([p_x,p_y,z]))<=diam:mask[int((p_y-origin[1])/spacing[1]),int((p_x-origin[0])/spacing[0])] = 1.0return (mask)luna_path = "E:/LUNA16/src/subset0/"
output_path = "E:/VSCode/lung/npy"
file_list = glob("E:\LUNA16\src\subset0/" + "*.mhd")# 获取数据中的每一行
def get_filename(case):global file_listfor f in file_list:if case in f:return (f)# 节点的位置
df_node = pd.read_csv('E:\LUNA16\src/annotations.csv')
df_node["file"] = df_node["seriesuid"].apply(get_filename)
# df_node: 文件的路径  example:E:\LUNA16\src\subset0\1.3.6.1.4.1.14519.5.2.1……
df_node = df_node.dropna()# 循环遍历图像文件
#fcount = 0
for fcount, img_file in enumerate(tqdm(file_list)):print("Getting mask for image file %s" % img_file.replace(luna_path, ""))mini_df = df_node[df_node["file"] == img_file]  # 得到所有结节if(len(mini_df) > 0):    #跳过没有结节的文件#读取数据itk_img = sitk.ReadImage(img_file)img_array = sitk.GetArrayFromImage(itk_img)   #(z,y,x)num_z,height,width = img_array.shape    #heightXwidth constitute the transverse plane  origin = np.array(itk_img.GetOrigin())  #世界坐标系下的x,y,z(mm)spacing = np.array(itk_img.GetSpacing()) #世界坐标中的体素间隔(mm)#遍历所有节点for node_idx,cur_row in mini_df.iterrows():node_x = cur_row["coordX"]node_y = cur_row["coordY"]node_z = cur_row["coordZ"]diam = cur_row["diameter_mm"]#保留三个切片imgs = np.ndarray([3,height,width],dtype=np.float32)masks = np.ndarray([3,height,width],dtype=np.uint8)center = np.array([node_x,node_y,node_z])  #结点中心v_center = np.rint((center-origin)/spacing) #体素坐标系的结点中心(x,y,z)for i,i_z in enumerate(np.arange(int(v_center[2])-1,int(v_center[2])+2).clip(0,num_z-1)):   #clip防止超出zmask = make_mask(center,diam,i_z*spacing[2]+origin[2],width,height,spacing,origin)masks[i] = mask#imgs[i] = img_array[i_z]imgs[i] = normalizePlanes(img_array[i_z])np.save(os.path.join(output_path,"images_%04d_%04d.npy" % (fcount, node_idx)),imgs)np.save(os.path.join(output_path,"masks_%04d_%04d.npy" % (fcount, node_idx)),masks)

1.2  获取肺实质

for img_file in file_list:imgs_to_process = np.load(img_file).astype(np.float64) print("on image",img_file)for i in range(len(imgs_to_process)):img = imgs_to_process[i]#标准化mean = np.mean(img)   #均值std = np.std(img)    #标准差img = (img-mean)/std#寻找肺部附近的平均像素,以重新调整过度曝光的图像middle = img[100:400,100:400]#使用Kmeans算法将前景(放射性不透明组织)和背景(放射性透明组织,即肺部)分离。#仅在图像中心进行此操作,以尽可能避免图像的非组织部分。kmeans = KMeans(n_clusters=2,n_init=10).fit(np.reshape(middle,[np.prod(middle.shape),1]))#np.prod 计算给定数组中所有元素的乘积#np.reshape 低维变高维  .flatten() 高维变1维centers = sorted(kmeans.cluster_centers_.flatten())threshold = np.mean(centers)thresh_img = np.where(img<threshold,1.0,0.0)  #阈值化图像,二值化处理#腐蚀和膨胀eroded = morphology.erosion(thresh_img,np.ones([4,4]))dilation = morphology.dilation(eroded,np.ones([10,10]))labels = measure.label(dilation)#对二值图像进行标记,标记连通区域label_vals = np.unique(labels)  #获取标记值的唯一值,即标记的数量regions = measure.regionprops(labels) #标记区域good_labels = []for prop in regions:B = prop.bbox   #边界框if B[2]-B[0]<475 and B[3]-B[1]<475 and B[0]>40 and B[2]<472:good_labels.append(prop.label)#np.ndarray 创建多维数组mask = np.ndarray([512,512],dtype=np.int8)mask[:] = 0  #肺部maskfor N in good_labels:mask = mask + np.where(labels==N,1,0)mask = morphology.dilation(mask,np.ones([10,10]))imgs_to_process[i] = masknp.save(img_file.replace("images","lungmask"),imgs_to_process)

1.3  获取xml和jpg文件

        通过遍历只含结节图像nodule_annos.xls,该文件在另一篇文章中生成。点此跳转

       这里也可以通过遍历官方文件annotations.csv获得

        获取结节图像的结节信息生成xml文件,生成xml文件的方法参考了这位博主的方法,可以去学习,生成不了也可以去下载该博主共享的数据集,点此跳转

        获取名称匹配lungmask图像获得jpg图像

count = 0
for i in range(len(img_list)):origin_img = np.load(img_list[i])lung_mask = np.load(lungmask_list[i])for j in range(len(origin_img)):img = origin_img[j]lung_img = lung_mask[j]segment = img * lung_imgsegment = Image.fromarray(segment)segment = segment.convert('RGB')segment.save("E:\VSCode\lung/" + '{:04d}.jpg'.format(count+1))count = count + 1

2、模型

YOLOV7

 3、训练

2023.4.14更

 

当然后续继续训练也是有很大的提升的,模型没有完全收敛

2023.5.15 更


有小伙伴私信问我说xml中的filename和图片名称对不上,具体的原因是因为我删掉了一部分肺实质分割不太好的分割图,所以导致最后出现不匹配的原因。
不过labelimg查看结节位置都是对的,训练起来也没有太大的影响

这里用无删减的数据集训练了一下

训练结果

检测结果

这篇关于使用Yolov7进行LUNA16的肺结节检测的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/836286

相关文章

springboot security使用jwt认证方式

《springbootsecurity使用jwt认证方式》:本文主要介绍springbootsecurity使用jwt认证方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地... 目录前言代码示例依赖定义mapper定义用户信息的实体beansecurity相关的类提供登录接口测试提供一

go中空接口的具体使用

《go中空接口的具体使用》空接口是一种特殊的接口类型,它不包含任何方法,本文主要介绍了go中空接口的具体使用,具有一定的参考价值,感兴趣的可以了解一下... 目录接口-空接口1. 什么是空接口?2. 如何使用空接口?第一,第二,第三,3. 空接口几个要注意的坑坑1:坑2:坑3:接口-空接口1. 什么是空接

springboot security快速使用示例详解

《springbootsecurity快速使用示例详解》:本文主要介绍springbootsecurity快速使用示例,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝... 目录创www.chinasem.cn建spring boot项目生成脚手架配置依赖接口示例代码项目结构启用s

Python如何使用__slots__实现节省内存和性能优化

《Python如何使用__slots__实现节省内存和性能优化》你有想过,一个小小的__slots__能让你的Python类内存消耗直接减半吗,没错,今天咱们要聊的就是这个让人眼前一亮的技巧,感兴趣的... 目录背景:内存吃得满满的类__slots__:你的内存管理小助手举个大概的例子:看看效果如何?1.

java中使用POI生成Excel并导出过程

《java中使用POI生成Excel并导出过程》:本文主要介绍java中使用POI生成Excel并导出过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录需求说明及实现方式需求完成通用代码版本1版本2结果展示type参数为atype参数为b总结注:本文章中代码均为

Spring Boot3虚拟线程的使用步骤详解

《SpringBoot3虚拟线程的使用步骤详解》虚拟线程是Java19中引入的一个新特性,旨在通过简化线程管理来提升应用程序的并发性能,:本文主要介绍SpringBoot3虚拟线程的使用步骤,... 目录问题根源分析解决方案验证验证实验实验1:未启用keep-alive实验2:启用keep-alive扩展建

使用Java实现通用树形结构构建工具类

《使用Java实现通用树形结构构建工具类》这篇文章主要为大家详细介绍了如何使用Java实现通用树形结构构建工具类,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录完整代码一、设计思想与核心功能二、核心实现原理1. 数据结构准备阶段2. 循环依赖检测算法3. 树形结构构建4. 搜索子

基于Flask框架添加多个AI模型的API并进行交互

《基于Flask框架添加多个AI模型的API并进行交互》:本文主要介绍如何基于Flask框架开发AI模型API管理系统,允许用户添加、删除不同AI模型的API密钥,感兴趣的可以了解下... 目录1. 概述2. 后端代码说明2.1 依赖库导入2.2 应用初始化2.3 API 存储字典2.4 路由函数2.5 应

GORM中Model和Table的区别及使用

《GORM中Model和Table的区别及使用》Model和Table是两种与数据库表交互的核心方法,但它们的用途和行为存在著差异,本文主要介绍了GORM中Model和Table的区别及使用,具有一... 目录1. Model 的作用与特点1.1 核心用途1.2 行为特点1.3 示例China编程代码2. Tab

SpringBoot使用OkHttp完成高效网络请求详解

《SpringBoot使用OkHttp完成高效网络请求详解》OkHttp是一个高效的HTTP客户端,支持同步和异步请求,且具备自动处理cookie、缓存和连接池等高级功能,下面我们来看看SpringB... 目录一、OkHttp 简介二、在 Spring Boot 中集成 OkHttp三、封装 OkHttp