学习:Mobile-Unet网络的肺结节图像分割方法

简介: 学习:Mobile-Unet网络的肺结节图像分割方法

前言


  针对医学图像中分辨率低、边缘模糊和感兴趣区域边界模糊导致分割不准确的问题,提出了一种新的Mobile-Unet网络肺结节图像分割方法。该方法首先用MobileNet中的bneck模块代替Unet网络的下采样部分,并从输入图像中提取特征;然后,根据Unet网络的连接方法,将下采样提取的特征融合到上采样部分;最后,利用训练好的网络得到分割结果。


  实验使用收集的肺结节数据集来训练和验证移动Unet。结果表明,新的移动Unet网络可以更准确地分割肺结节的位置,分割精度从原始Unet网络的85.00%提高到90.00%,相似系数(F1)仍然稳定在89.98%,证明了该方法的有效性



实验准备


项目整体实现步骤如下:

  1. 图像预处理
  2. 网络算法架构(Unet网络&MobileNetV3网络)
  3. 结合上述二者网络


数据集欲处理


  本文所使用的数据集图像格式为DICM 格式。数据集中共有20例病,这些病人共有400张CT图像,其中400多张CT图像可能一1到2个节点(一个节点可能含有多个切片),将有结节部分的图像400多张CT图像挑选部分组成数据集。


  为了使数据集可以完美应用到网络模型中,我们需要对数据集进行归一化处理,使用Unet网络时候需要将掩膜作为原始图像的标签,使用标注软件对节点位置进行人工标注生成掩膜图像。掩膜中含有2个像素点,分别是0和1;1代表节点所在的像素,0代表非节点所在的像素。

image.png


网络架构


  为了提高肺节点图像精度和效率,我们可以结合:Mobile-Unet网络模型,其中主干网络选用MobileNetV3轻量级网络,任务网络选用常用的Unet网络。


   该网络通过融合深层特征的方式提取表达性较强的特征,解决分割精度难以提升的问题;采用MobileNetV3网络特征提取可以降低模型复杂度、减少训练时的参数量,从而提高网络的收敛速度。



实验


   本文将数据集(217张图片) 划分为训练集、验证集和测试集,训练集用以训练模型,验证集用以对模型进行调优工作,测试集用以测试模型的性能。首先,取出后200张图片;然后,使用机器学习包中的split方法将图片划分为训练集和验证集,划分比例为9:1;最后,将图片前17张作为测试集。


    将训练集送入Mobile-Unet网络训练,训练过程的误差曲线如下图所示。 由下图可以看出:训练集和验证集的误差都随着步数的增加而下降,最后趋于稳定,并且误差降到 1%以下,说明该模型对偏差控制得很好;训练集和验证集稳定后的误差接近 说明该模型的方差比较小。

image.png

image.png



相关文章
|
6月前
|
机器学习/深度学习 算法 调度
14种智能算法优化BP神经网络(14种方法)实现数据预测分类研究(Matlab代码实现)
14种智能算法优化BP神经网络(14种方法)实现数据预测分类研究(Matlab代码实现)
489 0
|
7月前
|
机器学习/深度学习 算法 数据挖掘
没发论文的注意啦!重磅更新!GWO-BP-AdaBoost预测!灰狼优化、人工神经网络与AdaBoost集成学习算法预测研究(Matlab代码实现)
没发论文的注意啦!重磅更新!GWO-BP-AdaBoost预测!灰狼优化、人工神经网络与AdaBoost集成学习算法预测研究(Matlab代码实现)
223 0
|
6月前
|
JavaScript Java 大数据
基于python的网络课程在线学习交流系统
本研究聚焦网络课程在线学习交流系统,从社会、技术、教育三方面探讨其发展背景与意义。系统借助Java、Spring Boot、MySQL、Vue等技术实现,融合云计算、大数据与人工智能,推动教育公平与教学模式创新,具有重要理论价值与实践意义。
|
5月前
|
机器学习/深度学习 数据采集 边缘计算
基于灰色神经网络的预测方法
基于灰色神经网络的预测方法
333 0
|
6月前
|
算法 Python
【EI复现】考虑网络动态重构的分布式电源选址定容优化方法(Matlab代码实现)
【EI复现】考虑网络动态重构的分布式电源选址定容优化方法(Matlab代码实现)
207 0
|
7月前
|
机器学习/深度学习 数据采集 TensorFlow
基于CNN-GRU-Attention混合神经网络的负荷预测方法(Python代码实现)
基于CNN-GRU-Attention混合神经网络的负荷预测方法(Python代码实现)
386 0
|
8月前
|
存储 Linux 容器
【Container App】在容器中抓取网络包的方法
本文介绍在Azure Container App中安装tcpdump抓取网络包,并通过Storage Account上传抓包文件的方法。内容包括使用curl和nc测试外部接口连通性、长Ping端口、安装tcpdump、抓取网络包、以及通过crul命令上传文件至Azure Storage。适用于需要分析网络请求和排查网络问题的场景。
247 0
|
7月前
|
机器学习/深度学习 并行计算 算法
【图像分割】基于神经气体网络的图像分割与量化(Matlab代码实现)
【图像分割】基于神经气体网络的图像分割与量化(Matlab代码实现)
111 0
|
8月前
|
机器学习/深度学习 边缘计算 算法
基于BP神经网络的电池容量预测方法研究
基于BP神经网络的电池容量预测方法研究
|
监控 安全 网络安全
深入解析PDCERF:网络安全应急响应的六阶段方法
PDCERF是网络安全应急响应的六阶段方法,涵盖准备、检测、抑制、根除、恢复和跟进。本文详细解析各阶段目标与操作步骤,并附图例,助读者理解与应用,提升组织应对安全事件的能力。
2036 89

热门文章

最新文章