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

本文涉及的产品
视觉智能开放平台,视频资源包5000点
视觉智能开放平台,图像资源包5000点
视觉智能开放平台,分割抠图1万点
简介: 学习: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



相关文章
|
23天前
|
机器学习/深度学习 数据采集 人工智能
基于Huffman树的层次化Softmax:面向大规模神经网络的高效概率计算方法
层次化Softmax算法通过引入Huffman树结构,将传统Softmax的计算复杂度从线性降至对数级别,显著提升了大规模词汇表的训练效率。该算法不仅优化了计算效率,还在处理大规模离散分布问题上提供了新的思路。文章详细介绍了Huffman树的构建、节点编码、概率计算及基于Gensim的实现方法,并讨论了工程实现中的优化策略与应用实践。
65 15
基于Huffman树的层次化Softmax:面向大规模神经网络的高效概率计算方法
|
14天前
|
域名解析 缓存 网络协议
优化Lua-cURL:减少网络请求延迟的实用方法
优化Lua-cURL:减少网络请求延迟的实用方法
|
2月前
|
机器学习/深度学习 数据采集 算法
机器学习在医疗诊断中的前沿应用,包括神经网络、决策树和支持向量机等方法,及其在医学影像、疾病预测和基因数据分析中的具体应用
医疗诊断是医学的核心,其准确性和效率至关重要。本文探讨了机器学习在医疗诊断中的前沿应用,包括神经网络、决策树和支持向量机等方法,及其在医学影像、疾病预测和基因数据分析中的具体应用。文章还讨论了Python在构建机器学习模型中的作用,面临的挑战及应对策略,并展望了未来的发展趋势。
142 1
|
2月前
|
安全 算法 网络安全
量子计算与网络安全:保护数据的新方法
量子计算的崛起为网络安全带来了新的挑战和机遇。本文介绍了量子计算的基本原理,重点探讨了量子加密技术,如量子密钥分发(QKD)和量子签名,这些技术利用量子物理的特性,提供更高的安全性和可扩展性。未来,量子加密将在金融、政府通信等领域发挥重要作用,但仍需克服量子硬件不稳定性和算法优化等挑战。
|
2月前
|
编解码 安全 Linux
网络空间安全之一个WH的超前沿全栈技术深入学习之路(10-2):保姆级别教会你如何搭建白帽黑客渗透测试系统环境Kali——Liinux-Debian:就怕你学成黑客啦!)作者——LJS
保姆级别教会你如何搭建白帽黑客渗透测试系统环境Kali以及常见的报错及对应解决方案、常用Kali功能简便化以及详解如何具体实现
|
2月前
|
安全 网络协议 算法
网络空间安全之一个WH的超前沿全栈技术深入学习之路(8-1):主动信息收集之ping、Nmap 就怕你学成黑客啦!
网络空间安全之一个WH的超前沿全栈技术深入学习之路(8-1):主动信息收集之ping、Nmap 就怕你学成黑客啦!
|
2月前
|
网络协议 安全 NoSQL
网络空间安全之一个WH的超前沿全栈技术深入学习之路(8-2):scapy 定制 ARP 协议 、使用 nmap 进行僵尸扫描-实战演练、就怕你学成黑客啦!
scapy 定制 ARP 协议 、使用 nmap 进行僵尸扫描-实战演练等具体操作详解步骤;精典图示举例说明、注意点及常见报错问题所对应的解决方法IKUN和I原们你这要是学不会我直接退出江湖;好吧!!!
网络空间安全之一个WH的超前沿全栈技术深入学习之路(8-2):scapy 定制 ARP 协议 、使用 nmap 进行僵尸扫描-实战演练、就怕你学成黑客啦!
|
2月前
|
人工智能 安全 Linux
网络空间安全之一个WH的超前沿全栈技术深入学习之路(4-2):渗透测试行业术语扫盲完结:就怕你学成黑客啦!)作者——LJS
网络空间安全之一个WH的超前沿全栈技术深入学习之路(4-2):渗透测试行业术语扫盲完结:就怕你学成黑客啦!)作者——LJS
|
2月前
|
安全 大数据 Linux
网络空间安全之一个WH的超前沿全栈技术深入学习之路(3-2):渗透测试行业术语扫盲)作者——LJS
网络空间安全之一个WH的超前沿全栈技术深入学习之路(3-2):渗透测试行业术语扫盲)作者——LJS

热门文章

最新文章