使用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

相关文章

使用Python实现图像LBP特征提取的操作方法

《使用Python实现图像LBP特征提取的操作方法》LBP特征叫做局部二值模式,常用于纹理特征提取,并在纹理分类中具有较强的区分能力,本文给大家介绍了如何使用Python实现图像LBP特征提取的操作方... 目录一、LBP特征介绍二、LBP特征描述三、一些改进版本的LBP1.圆形LBP算子2.旋转不变的LB

Maven的使用和配置国内源的保姆级教程

《Maven的使用和配置国内源的保姆级教程》Maven是⼀个项目管理工具,基于POM(ProjectObjectModel,项目对象模型)的概念,Maven可以通过一小段描述信息来管理项目的构建,报告... 目录1. 什么是Maven?2.创建⼀个Maven项目3.Maven 核心功能4.使用Maven H

Python中__init__方法使用的深度解析

《Python中__init__方法使用的深度解析》在Python的面向对象编程(OOP)体系中,__init__方法如同建造房屋时的奠基仪式——它定义了对象诞生时的初始状态,下面我们就来深入了解下_... 目录一、__init__的基因图谱二、初始化过程的魔法时刻继承链中的初始化顺序self参数的奥秘默认

SpringBoot使用GZIP压缩反回数据问题

《SpringBoot使用GZIP压缩反回数据问题》:本文主要介绍SpringBoot使用GZIP压缩反回数据问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录SpringBoot使用GZIP压缩反回数据1、初识gzip2、gzip是什么,可以干什么?3、Spr

Spring Boot 集成 Quartz并使用Cron 表达式实现定时任务

《SpringBoot集成Quartz并使用Cron表达式实现定时任务》本篇文章介绍了如何在SpringBoot中集成Quartz进行定时任务调度,并通过Cron表达式控制任务... 目录前言1. 添加 Quartz 依赖2. 创建 Quartz 任务3. 配置 Quartz 任务调度4. 启动 Sprin

Linux下如何使用C++获取硬件信息

《Linux下如何使用C++获取硬件信息》这篇文章主要为大家详细介绍了如何使用C++实现获取CPU,主板,磁盘,BIOS信息等硬件信息,文中的示例代码讲解详细,感兴趣的小伙伴可以了解下... 目录方法获取CPU信息:读取"/proc/cpuinfo"文件获取磁盘信息:读取"/proc/diskstats"文

Java使用SLF4J记录不同级别日志的示例详解

《Java使用SLF4J记录不同级别日志的示例详解》SLF4J是一个简单的日志门面,它允许在运行时选择不同的日志实现,这篇文章主要为大家详细介绍了如何使用SLF4J记录不同级别日志,感兴趣的可以了解下... 目录一、SLF4J简介二、添加依赖三、配置Logback四、记录不同级别的日志五、总结一、SLF4J

使用Python实现一个优雅的异步定时器

《使用Python实现一个优雅的异步定时器》在Python中实现定时器功能是一个常见需求,尤其是在需要周期性执行任务的场景下,本文给大家介绍了基于asyncio和threading模块,可扩展的异步定... 目录需求背景代码1. 单例事件循环的实现2. 事件循环的运行与关闭3. 定时器核心逻辑4. 启动与停

如何使用Nginx配置将80端口重定向到443端口

《如何使用Nginx配置将80端口重定向到443端口》这篇文章主要为大家详细介绍了如何将Nginx配置为将HTTP(80端口)请求重定向到HTTPS(443端口),文中的示例代码讲解详细,有需要的小伙... 目录1. 创建或编辑Nginx配置文件2. 配置HTTP重定向到HTTPS3. 配置HTTPS服务器

Java使用ANTLR4对Lua脚本语法校验详解

《Java使用ANTLR4对Lua脚本语法校验详解》ANTLR是一个强大的解析器生成器,用于读取、处理、执行或翻译结构化文本或二进制文件,下面就跟随小编一起看看Java如何使用ANTLR4对Lua脚本... 目录什么是ANTLR?第一个例子ANTLR4 的工作流程Lua脚本语法校验准备一个Lua Gramm