CN113920108A - 一种对用于处理细胞图像的U-Net模型进行训练的训练方法 - Google Patents
一种对用于处理细胞图像的U-Net模型进行训练的训练方法 Download PDFInfo
- Publication number
- CN113920108A CN113920108A CN202111274845.6A CN202111274845A CN113920108A CN 113920108 A CN113920108 A CN 113920108A CN 202111274845 A CN202111274845 A CN 202111274845A CN 113920108 A CN113920108 A CN 113920108A
- Authority
- CN
- China
- Prior art keywords
- image
- training
- net model
- training set
- cell
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 238000000034 method Methods 0.000 title claims abstract description 42
- 238000002372 labelling Methods 0.000 claims abstract description 11
- 230000004913 activation Effects 0.000 claims description 15
- 238000011176 pooling Methods 0.000 claims description 9
- 238000003384 imaging method Methods 0.000 claims description 8
- 230000003628 erosive effect Effects 0.000 claims description 7
- 230000008569 process Effects 0.000 claims description 6
- 238000003672 processing method Methods 0.000 claims description 6
- 238000005070 sampling Methods 0.000 claims description 6
- 230000003044 adaptive effect Effects 0.000 claims description 5
- 230000010339 dilation Effects 0.000 claims description 5
- 230000000877 morphologic effect Effects 0.000 claims description 5
- 230000011218 segmentation Effects 0.000 abstract description 9
- 238000003709 image segmentation Methods 0.000 abstract description 7
- 210000004027 cell Anatomy 0.000 description 61
- 230000006870 function Effects 0.000 description 12
- 210000002540 macrophage Anatomy 0.000 description 10
- 210000003743 erythrocyte Anatomy 0.000 description 8
- 238000013136 deep learning model Methods 0.000 description 4
- 230000000694 effects Effects 0.000 description 4
- 238000013527 convolutional neural network Methods 0.000 description 3
- 238000001514 detection method Methods 0.000 description 3
- 238000013528 artificial neural network Methods 0.000 description 2
- 238000010586 diagram Methods 0.000 description 2
- 230000001464 adherent effect Effects 0.000 description 1
- 230000004075 alteration Effects 0.000 description 1
- 238000000149 argon plasma sintering Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 210000004369 blood Anatomy 0.000 description 1
- 239000008280 blood Substances 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000003745 diagnosis Methods 0.000 description 1
- 201000010099 disease Diseases 0.000 description 1
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 description 1
- 238000000605 extraction Methods 0.000 description 1
- 238000001914 filtration Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/0002—Inspection of images, e.g. flaw detection
- G06T7/0012—Biomedical image inspection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T3/00—Geometric image transformations in the plane of the image
- G06T3/40—Scaling of whole images or parts thereof, e.g. expanding or contracting
- G06T3/4007—Scaling of whole images or parts thereof, e.g. expanding or contracting based on interpolation, e.g. bilinear interpolation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/20—Image enhancement or restoration using local operators
- G06T5/30—Erosion or dilatation, e.g. thinning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/70—Denoising; Smoothing
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/11—Region-based segmentation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/12—Edge-based segmentation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/13—Edge detection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/136—Segmentation; Edge detection involving thresholding
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/10—Image acquisition modality
- G06T2207/10056—Microscopic image
- G06T2207/10061—Microscopic image from scanning electron microscope
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20036—Morphological image processing
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- General Health & Medical Sciences (AREA)
- Health & Medical Sciences (AREA)
- Biophysics (AREA)
- Software Systems (AREA)
- Evolutionary Computation (AREA)
- Computational Linguistics (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Biomedical Technology (AREA)
- Mathematical Physics (AREA)
- Data Mining & Analysis (AREA)
- Artificial Intelligence (AREA)
- Life Sciences & Earth Sciences (AREA)
- Medical Informatics (AREA)
- Nuclear Medicine, Radiotherapy & Molecular Imaging (AREA)
- Radiology & Medical Imaging (AREA)
- Quality & Reliability (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种对用于处理细胞图像的U‑Net模型进行训练的训练方法,该训练方法基于预先获得的标注细胞图像数据集对U‑Net模型进行训练,经过训练的U‑Net模型能够对细胞图像进行准确分割,分割后的轮廓边缘清晰,有效提高了对细胞图像,尤其是混合种类细胞图像进行分割和/或计数的准确性,有效解决了现有技术中存在的图像标注成本高、细胞图像分割困难以及细胞计数操作复杂等难题。
Description
技术领域
本发明属于计算机图像处理技术领域,具体涉及一种对用于处理细胞图像的U-Net模型进行训练的训练方法。
背景技术
目前许多疾病的诊断非常依赖于病人血液内细胞数量的计数结果,而临床上常用的细胞计数手段基本都是基于细胞分析仪,一般根据两种计数原理:电阻抗法,当不同大小的细胞通过检测通道时,会产生不同的电压脉冲信号;光散射法,当不同波长的光照到不同细胞时,会得到不同的衍射角。但是此种仪器具有操作复杂、成本较高、耗时等缺点。
图像分割是根据图像的特征将图像划分成若干特定区域,是一种以像素为单位标记图像中的感兴趣区域的图像处理过程。对于单一种类的细胞分割,传统方法主要是基于二值化或者边缘特征,对于粘连细胞通常会采用分水岭算法进行处理。但是由于细胞图像拍摄的因素以及细胞生存环境的复杂性,传统图像处理算法不能完全胜任混合种类细胞的分割。
近年来,基于卷积神经网络的方法在计算机视觉领域被广泛应用于图像分类、目标检测、图像去噪、语义分割等任务中。不同于传统图像处理方法,深度学习可以从细胞图像中学习到不同细胞的特征,从而能够对不同的细胞进行分类。其中,全卷积神经网络可以获得较好的图像分割性能。全卷积神经网络去掉了传统卷积神经网络的全连接层,并使用上采样操作实现端到端(即像素到像素)图像分割。
U-Net模型是一个具有编解码器结构的全卷积神经网络,该模型中的编码器模块用于特征提取,解码器模块用于恢复原始图像分辨率的特征图。U-Net模型通常在小尺度图像数据集上取得较好的分割效果,由于其编解码器结构,适合于医学图像分割。U-Net模型使用跳过连接将编码器模块的浅层特征图与解码器模块的深层特征图结合起来,以恢复细粒度的对象细节。但是为了训练深度学习模型,需要大量具有像素级标签的数据,然而在专家的协助下手动标记数据是一个费劲且昂贵的过程。
有鉴于此,如何进一步降低图像标注成本、细胞图像分割难度以及细胞计数操作复杂度,成为目前亟待解决的技术课题。
发明内容
为克服上述现有技术中存在的缺陷和不足,本发明提供了一种对用于处理细胞图像的U-Net模型进行训练的训练方法,该训练方法基于预先获得的标注细胞图像数据集对U-Net模型进行训练,经过训练的U-Net模型能够对细胞图像进行准确分割,分割后的轮廓边缘清晰,有效提高对细胞图像,尤其是混合种类细胞图像进行分割和/或计数的准确性。
U-Net模型是一种常见的用于医学图像处理的深度学习模型,该模型主要有三个特点:(1)用于下采样的编码器路径,(2)用于上采样的解码器路径,(3)用于结合编码器和解码器信息的跳连接。但深度学习模型想要获得良好的训练效果,往往需要大量的人工标注的训练集,而训练集的标注成本较高。本发明所提供的该种训练方法使用传统图像处理方法预先获得标注细胞图像数据集,并使用其对U-Net模型进行训练,这样可以在有效降低成本的同时能够获得足够的训练数据来进行U-Net模型的训练,优势明显,能有效解决现有技术中存在的图像标注成本高、细胞图像分割困难以及细胞计数操作复杂等难题。
为达上述目的,本发明所提供的该种训练方法,其具体包括以下步骤:
S1、获取单种类细胞图像并制作图像训练集;
S2、使用图像处理算法处理上述步骤S1中所制作的图像训练集中的每一张图像并得到该图像对应的标签掩膜图,最终形成图像训练集对应的标签掩膜图训练集;
S3、使用上述步骤S2所获得的标签掩膜图训练集对U-Net模型进行训练;
S4、获取混合种类细胞图像并制作微调数据集;
S5、使用上述步骤S4所制作的微调数据集对经上述步骤S3训练过的U-Net模型进行微调式训练。
在本发明的一个具体实施例中,上述步骤S1可包括以下步骤:
S11、使用成像设备获取所需的单种类细胞图像;
S12、将上述步骤S11中所获取的单种类细胞图像切割成若干固定大小的图像,最终汇集成图像训练集。
本领域技术人员可以理解的是,上述步骤S11中使用的成像设备可以是任何适用的现有成像设备,例如可连接到显微镜上的高清摄像机、高清摄像头等。
在本发明的一较佳实施例中,上述步骤S12中,可以将图像切割成固定的通行大小512×512×3。
本领域技术人员可以理解的是,上述步骤S2中所使用的图像处理算法可以是任何适用的现有图像处理算法,例如可以包括高斯滤波、自适应阈值算法、铃木轮廓检测算法、形态学腐蚀膨胀操作等。
有鉴于此,在本发明的一个具体实施例中,上述步骤S2可包括以下步骤:
S21、使用高斯滤波处理上述步骤S1中所制作的图像训练集中的每一张图像,去除图像中的噪音;
S22、使用自适应阈值算法处理经上述步骤S21处理过的图像训练集中的每一张图像,得到图像二值化的标签;
S23、使用铃木轮廓追踪算法对经上述步骤S22处理过的图像训练集中的每一张图像进行图像边缘检测;
S24、对经上述步骤S23处理过的图像训练集中的每一张图像的图像边缘轮廓进行填充,得到该图像对应的标签掩膜图,最终汇集成图像训练集对应的标签掩膜图训练集。
本领域技术人员可以理解的是,上述步骤21中所去除的噪音可以是图像捕获过程中产生的各类噪音。
在本发明的一较佳实施例中,为提高精度,上述步骤24中,在对图像边缘轮廓进行填充之前可以移除面积较小的轮廓。
在本发明的一较佳实施例中,上述步骤24中所获得的标签掩膜图大小仍然保持为512×512×3。
在本发明的一较佳实施例中,为提高精度,对于体积较大的细胞来说,例如巨噬细胞,上述步骤24中在获得图像对应的标签掩膜图之前对图像追加形态学腐蚀膨胀操作,优选地,可以先进行腐蚀操作,然后再进行膨胀操作,如此反复进行2次。但本领域技术人员也能理解的是,前述形态学腐蚀膨胀操作不是必须的。
在本发明的一个具体实施例中,上述步骤S3可包括以下步骤:
S31、设置卷积核大小为3×3、步长和填充为1的卷积层;
S32、设置卷积核大小为2×2和4×4两种步长为2的池化层;
S33、设置2倍和4倍的最近邻插值层;
S34、设置卷积核大小为1×1的卷积层;
S35、将上述步骤S2获得的标签掩膜图训练集输出送入U-Net模型,对标签掩膜图训练集中的每一张图像进行四次下采样;
S36、将上述步骤S2获得的标签掩膜图训练集输出送入U-Net模型,对标签掩膜图训练集中的每一张图像进行四次上采样,并结合上述步骤35获得的下采样图像将对应层中大小相同的输出进行拼接;
S37、对经过上述步骤36处理过的上采样图像进行1×1的卷积操作,使用激活函数得到标签掩膜图训练集中的每一张图像的最终预测图;
S38、使用上述步骤37中所获得的最终预测图与其所对应图像的真实值之间的差异作为损失,基于该损失进一步更新U-Net模型,实现对U-Net模型的训练。
本领域技术人员可以理解的是,上述步骤S3中,对U-Net模型的训练可经过多次迭代以使U-Net模型达到理想性能,迭代次数越多,U-Net模型所能达到的性能越理想,但这也会导致成本的升高,训练时的学习率和具体迭代次数可以由本领域技术人员根据实际情况进行设置。
其中,上述步骤S35中,前三次下采样中的每一次均先经过两次3×3的卷积,其中每次卷积后都使用激活函数进行激活,然后再经过一次2×2的池化操作,第四次下采样仅经过一次4×4的池化操作;上述步骤S36中,第一次上采样经过两次3×3的卷积,其中卷积后使用激活函数进行激活,然后再经过4倍的最近邻插值操作,后三次上采样中的每一次均经过2倍的最近邻插值操作。
本领域技术人员可以理解的是,上述步骤S35和S36中使用的激活函数可以是任何适用的现有激活函数,例如ReLU激活函数。
本领域技术人员可以理解的是,上述步骤S37中使用的激活函数可以是任何适用的现有激活函数,例如softmax激活函数。
在本发明的一较佳实施例中,上述步骤S37中得到的最终预测图大小是512×512×3,与上述步骤S2中获得的标签掩膜图大小相同。
在本发明的一个具体实施例中,上述步骤S4可包括以下步骤:
S41、使用成像设备获取所需的混合种类细胞图像;
S42、将上述步骤S41中获取的混合种类细胞图像切割成若干固定大小的图像;
S43、对上述步骤S42中切割后获得的每一张图像进行人工标注,并汇集成微调数据集。
本领域技术人员可以理解的是,上述步骤S41中使用的成像设备可以是任何适用的现有成像设备,例如可连接到显微镜上的高清摄像机、高清摄像头等。
在本发明的一较佳实施例中,上述步骤S42中,可以将图像切割成固定的通行大小512×512×3。
本领域技术人员可以理解的是,上述步骤S43中,可以使用任何适用的现有图像标注工具进行人工标注,例如深度学习图像标注工具labelme。
本领域技术人员可以理解的是,上述步骤S43中,制作的微调数据集中图像的张数越多,用其对U-Net模型进行微调训练的效果就越好,但这也会导致成本的升高,具体张数可以由本领域技术人员根据实际情况进行设置。
在本发明的一个具体实施例中,上述步骤S5中,使用上述步骤S4所制作的微调数据集对经上述步骤S3训练过的U-Net模型进行微调式训练,微调式训练时的学习率和迭代次数可以由本领域技术人员根据实际情况进行设置。
本领域技术人员可以理解的是,这里的“微调式训练”具体是指使用步骤S4中由混合种类细胞图像生成的微调数据集进一步训练步骤S3中训练过的U-Net模型,以期望进一步提升其性能。
进一步地,本发明继而提供一种细胞图像的处理方法,该处理方法使用经训练的上述U-Net模型对细胞图像进行分割和/或计数。
本领域技术人员可以理解的是,上述处理方法在处理混合细胞图像时尤其有效。
在本发明的一个具体实施例中,为提高准确度,优选地,上述处理方法中,计数时需要根据分割区域的面积进行判断,如果面积过小则不计数,面积过大则需要将其计数两次。
与先用技术相比,本发明具有以下有益的效果:
本发明所提供的该种U-Net模型的训练方法,能有效利用传统图像处理算法高效自动的特点对细胞图像进行自动标注,实现了数据集的自动标注,极大地节省了人工成本;基于U-Net模型实现了对混合种类细胞图像的有效分割和/或计数,可大大减少临床中细胞计数的成本。
附图说明
本发明上述的和/或附加的方面和优点从下面结合附图对实施例的描述中将变得明显和容易理解。在附图中,除非另外规定,否则贯穿多个附图相同的附图标记表示相同或相似的部件或元素。这些附图不一定是按照比例绘制的。应该理解,这些附图仅描绘了根据本发明公开的一些实施方式,用来提供对本发明的进一步理解,构成本申请的一部分,而不应将其视为是对本发明范围的限制。其中:
图1为本发明提供的训练方法的流程示意图;
图2为本发明的训练方法所使用的一种单种类细胞图像,其中,(a)为原始的红细胞图像,(b)为处理后得到的红细胞的标签掩膜图;
图3为本发明的训练方法所使用的另一种单种类细胞图像,其中,(a)为原始的巨噬细胞图像,(b)为处理后得到的巨噬细胞的标签掩膜图;
图4为使用本发明提供的训练方法训练后的U-Net模型的结构示意图;
图5为使用U-Net模型对混合种类细胞图像进行分割和/或计数的结果示意图,其中,(a)为原始的混合种类细胞图像,(b)为使用仅经单种类细胞图像数据训练的U-Net模型的处理结果,(c)为使用本发明的训练方法所训练的U-Net模型的处理结果,(d)为基于(c)中的处理结果最终获得的识别并计数的结果。
具体实施方式
为了使本发明的目的、技术方案及优点更加清楚明白,以下结合附图及具体实施例,对本发明做进一步详细说明。应当理解,此处所描述的具体实施例仅用于解释本发明,而不能理解为对本发明的限制。
为了提升深度学习模型在细胞图像处理方面的性能,本发明提供一种对用于处理细胞图像的U-Net模型进行训练的训练方法,下面参考附图1-4描述该训练方法。其中,图1示出了该训练方法的具体流程,如图1所示,该训练方法具体包括以下步骤:
S1、获取单种类细胞图像并制作图像训练集;
S2、使用图像处理算法处理上述步骤S1中所制作的图像训练集中的每一张图像并得到该图像对应的标签掩膜图,最终形成图像训练集对应的标签掩膜图训练集;
S3、使用上述步骤S2所获得的标签掩膜图训练集对U-Net模型进行训练;
S4、获取混合种类细胞图像并制作微调数据集;
S5、使用上述步骤S4所制作的微调数据集对经上述步骤S3训练过的U-Net模型进行微调式训练。
在本实施例中,上述步骤S1包括步骤:S11、使用连接到显微镜上的高清摄像机获取所需的单种类细胞图像;S12、将步骤S11中所获取的单种类细胞图像切割成若干固定大小为512×512×3的图像,最终汇集成图像训练集。
这里,步骤S11中所获取的单种类细胞图像具体为红细胞图像或巨噬细胞图像,其中,图2中的(a)示出了获取的原始的红细胞图像,图3中的(a)示出了获取的原始的巨噬细胞图像。
在本实施例中,上述步骤S2包括步骤:S21、使用高斯滤波处理上述步骤S1中所制作的图像训练集中的每一张图像,去除图像中的噪音;其中,高斯滤波核的尺寸设定为3×3,行与列的方向上的标准偏差设置为0.8;S22、使用自适应阈值算法处理经上述步骤S21处理过的图像训练集中的每一张图像,得到图像二值化的标签;其中,进行区域自适应阈值处理时的滑动窗口的尺寸设定为3×3;S23、使用铃木轮廓追踪算法对经上述步骤S22处理过的图像训练集中的每一张图像进行图像边缘检测;S24、对经上述步骤S23处理过的图像训练集中的每一张图像的图像边缘轮廓进行填充,得到该图像对应的大小为512×512×3的标签掩膜图,最终汇集成图像训练集对应的标签掩膜图训练集。
优选地,为提高精度,上述步骤24中,在对图像边缘轮廓进行填充之前移除面积较小的轮廓。
优选地,为提高精度,针对巨噬细胞图像,上述步骤24中在获得图像对应的标签掩膜图之前对图像追加形态学腐蚀膨胀操作,具体为,先进行腐蚀操作,然后再进行膨胀操作,如此反复进行2次。
其中,图2中的(b)示出了处理后得到的红细胞的标签掩膜图,图3中的(b)示出了处理后得到的巨噬细胞的标签掩膜图。
在本实施例中,上述步骤S3包括步骤:S31、设置卷积核大小为3×3、步长和填充为1的卷积层;S32、设置卷积核大小为2×2和4×4两种步长为2的池化层;S33、设置2倍和4倍的最近邻插值层;S34、设置卷积核大小为1×1的卷积层;S35、将上述步骤S2获得的标签掩膜图训练集输出送入U-Net模型,对标签掩膜图训练集中的每一张图像进行四次下采样;S36、将上述步骤S2获得的标签掩膜图训练集输出送入U-Net模型,对标签掩膜图训练集中的每一张图像进行四次上采样,并结合上述步骤35获得的下采样图像将对应层中大小相同的输出进行拼接;S37、对经过上述步骤36处理过的上采样图像进行1×1的卷积操作,使用softmax激活函数得到标签掩膜图训练集中的每一张图像的最终预测图,该最终预测图的大小为512×512×3;S38、使用上述步骤37中所获得的最终预测图与其所对应图像的真实值之间的差异作为损失,基于该损失进一步更新U-Net模型,实现对U-Net模型的训练。
为使U-Net模型达到理想的性能,上述步骤S3中,对U-Net模型的训练需经过多次迭代完成,将训练时的学习率设定为0.001,迭代次数为200,根据实践经验,200代后损失基本稳定。
其中,上述步骤S35中,前三次下采样中的每一次均先经过两次3×3的卷积,其中每次卷积后都使用ReLU激活函数进行激活,然后再经过一次2×2的池化操作,第四次下采样仅经过一次4×4的池化操作;上述步骤S36中,第一次上采样经过两次3×3的卷积,其中卷积后使用ReLU激活函数进行激活,然后再经过4倍的最近邻插值操作,后三次上采样中的每一次均经过2倍的最近邻插值操作。
在本实施例中,上述步骤S4包括步骤:S41、使用连接到显微镜上的高清摄像机获取所需的混合种类细胞图像;S42、将上述步骤S41中获取的混合种类细胞图像切割成50张固定大小为512×512×3的图像;S43、对上述步骤S42中切割后获得的50张图像中的每一张使用深度学习图像标注工具labelme进行人工标注,并汇集成微调数据集。
这里,步骤S41中所获取的混合类细胞图像具体为红细胞和巨噬细胞的图像。
在本实施例中,上述步骤S5具体为,将微调式训练时的学习率设定为0.0005,迭代次数为100,使用上述步骤S4所制作的微调数据集对经上述步骤S3训练过的U-Net模型进行微调式训练。其中,图4示出了训练后的U-Net模型的具体结构。
进一步地,本发明继而提供一种细胞图像的处理方法,该处理方法使用经训练的上述U-Net模型对包含有红细胞和巨噬细胞的混合种类细胞图像进行分割和/或计数。
最后,对直接初步训练过的U-Net模型和微调式训练过的U-Net模型的处理效果进行对比,并从分割结果的准确性、细节信息的完整性和边缘轮廓的清晰性进行评价。其中,图5中的(a)示出了原始的混合种类细胞图像,图像中包含有红细胞和巨噬细胞;(b)示出了使用仅经单种类细胞图像数据训练的U-Net模型的处理结果;(c)则示出了使用本发明的训练方法所训练的U-Net模型的处理结果,从图中可以看到,分割效果识别准确,边缘清晰,细节信息完整;(d)则示出了基于(c)中的处理结果最终获得的识别并计数的结果,可见,混合种类细胞图像中的每个细胞均被清晰地识别出来,据此可以获得准确的细胞群数量。
尽管上面已经示出和描述了本发明的实施例,可以理解的是,上述实施例是示例性的,不能理解为对本发明的限制,本领域的普通技术人员在不脱离本发明的原理和宗旨的情况下在本发明的范围内可以对上述实施例进行变化、修改、替换和变型。
Claims (10)
1.一种对用于处理细胞图像的U-Net模型进行训练的训练方法,其特征在于,该训练方法包括以下步骤:
S1、获取单种类细胞图像并制作图像训练集;
S2、使用图像处理算法处理上述步骤S1中所制作的所述图像训练集中的每一张图像并得到该图像对应的标签掩膜图,最终形成所述图像训练集对应的标签掩膜图训练集;
S3、使用上述步骤S2所获得的所述标签掩膜图训练集对U-Net模型进行训练;
S4、获取混合种类细胞图像并制作微调数据集;
S5、使用上述步骤S4所制作的所述微调数据集对经上述步骤S3训练过的U-Net模型进行微调式训练。
2.根据权利要求1所述的训练方法,其特征在于,上述步骤S1包括以下步骤:
S11、使用成像设备获取所述单种类细胞图像;
S12、将上述步骤S11中所获取的所述单种类细胞图像切割成若干固定大小的图像,最终汇集成所述图像训练集。
3.根据权利要求1所述的训练方法,其特征在于,上述步骤S2包括以下步骤:
S21、使用高斯滤波处理上述步骤S1中所制作的所述图像训练集中的每一张图像,去除图像中的噪音;
S22、使用自适应阈值算法处理经上述步骤S21处理过的所述图像训练集中的每一张图像,得到图像二值化的标签;
S23、使用铃木轮廓追踪算法对经上述步骤S22处理过的所述图像训练集中的每一张图像进行图像边缘检测;
S24、对经上述步骤S23处理过的所述图像训练集中的每一张图像的图像边缘轮廓进行填充,得到该图像对应的标签掩膜图,最终汇集成所述图像训练集对应的所述标签掩膜图训练集。
4.根据权利要求3所述的训练方法,其特征在于,上述步骤24中,在对图像边缘轮廓进行填充之前移除面积较小的轮廓。
5.根据权利要求3所述的训练方法,其特征在于,上述步骤24中,在获得图像对应的标签掩膜图之前对图像追加形态学腐蚀膨胀操作。
6.根据权利要求1所述的训练方法,其特征在于,上述步骤S3包括以下步骤:
S31、设置卷积核大小为3×3、步长和填充为1的卷积层;
S32、设置卷积核大小为2×2和4×4两种步长为2的池化层;
S33、设置2倍和4倍的最近邻插值层;
S34、设置卷积核大小为1×1的卷积层;
S35、将上述步骤S2获得的所述标签掩膜图训练集输出送入U-Net模型,对所述标签掩膜图训练集中的每一张图像进行四次下采样;
S36、将上述步骤S2获得的所述标签掩膜图训练集输出送入U-Net模型,对所述标签掩膜图训练集中的每一张图像进行四次上采样,并结合上述步骤35获得的下采样图像将对应层中大小相同的输出进行拼接;
S37、对经过上述步骤36处理过的上采样图像进行1×1的卷积操作,使用激活函数得到所述标签掩膜图训练集中的每一张图像的最终预测图;
S38、使用上述步骤37中所获得的所述最终预测图与其所对应图像的真实值之间的差异作为损失,基于该损失进一步更新U-Net模型,实现对U-Net模型的训练。
7.根据权利要求6所述的训练方法,其特征在于,上述步骤S35中,前三次下采样中的每一次均先经过两次3×3的卷积,其中每次卷积后都使用激活函数进行激活,然后再经过一次2×2的池化操作,第四次下采样仅经过一次4×4的池化操作;上述步骤S36中,第一次上采样经过两次3×3的卷积,其中卷积后使用激活函数进行激活,然后再经过4倍的最近邻插值操作,后三次上采样中的每一次均经过2倍的最近邻插值操作。
8.根据权利要求1所述的训练方法,其特征在于,上述步骤S4包括以下步骤:
S41、使用成像设备获取所述混合种类细胞图像;
S42、将上述步骤S41中获取的所述混合种类细胞图像切割成若干固定大小的图像;
S43、对上述步骤S42中切割后获得的每一张图像进行人工标注,并汇集成所述微调数据集。
9.一种细胞图像的处理方法,其特征在于,该处理方法使用U-Net模型对所述细胞图像进行分割和/或计数,其中,所述U-Net模型为使用根据1-8中任意一项所述的训练方法所训练过的U-Net模型。
10.根据权利要求9所述的处理方法,其特征在于,所述细胞图像为混合种类细胞图像。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111274845.6A CN113920108A (zh) | 2021-10-29 | 2021-10-29 | 一种对用于处理细胞图像的U-Net模型进行训练的训练方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111274845.6A CN113920108A (zh) | 2021-10-29 | 2021-10-29 | 一种对用于处理细胞图像的U-Net模型进行训练的训练方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113920108A true CN113920108A (zh) | 2022-01-11 |
Family
ID=79243940
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202111274845.6A Pending CN113920108A (zh) | 2021-10-29 | 2021-10-29 | 一种对用于处理细胞图像的U-Net模型进行训练的训练方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113920108A (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114972968A (zh) * | 2022-05-19 | 2022-08-30 | 长春市大众物流装配有限责任公司 | 基于多重神经网络的托盘识别和位姿估计方法 |
CN116012838A (zh) * | 2022-12-30 | 2023-04-25 | 创芯国际生物科技(广州)有限公司 | 基于人工智能的类器官活性识别方法及系统 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109215041A (zh) * | 2018-08-17 | 2019-01-15 | 上海交通大学医学院附属第九人民医院 | 一种全自动骨盆肿瘤分割方法及系统、存储介质及终端 |
WO2020160839A1 (en) * | 2019-02-08 | 2020-08-13 | Carl Zeiss Meditec Ag | Segmentation and classification of geographic atrophy patterns in patients with age related macular degeneration in widefield autofluorescence images |
US20210019889A1 (en) * | 2017-01-27 | 2021-01-21 | Agfa Healthcare Nv | Multi-class image segmentation method |
CN113192077A (zh) * | 2021-04-15 | 2021-07-30 | 华中科技大学 | 一种细胞及区域层次的病理图自动分类方法及系统 |
CN113205528A (zh) * | 2021-04-02 | 2021-08-03 | 上海慧虎信息科技有限公司 | 医学影像分割模型训练方法、分割方法及装置 |
-
2021
- 2021-10-29 CN CN202111274845.6A patent/CN113920108A/zh active Pending
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20210019889A1 (en) * | 2017-01-27 | 2021-01-21 | Agfa Healthcare Nv | Multi-class image segmentation method |
CN109215041A (zh) * | 2018-08-17 | 2019-01-15 | 上海交通大学医学院附属第九人民医院 | 一种全自动骨盆肿瘤分割方法及系统、存储介质及终端 |
WO2020160839A1 (en) * | 2019-02-08 | 2020-08-13 | Carl Zeiss Meditec Ag | Segmentation and classification of geographic atrophy patterns in patients with age related macular degeneration in widefield autofluorescence images |
CN113205528A (zh) * | 2021-04-02 | 2021-08-03 | 上海慧虎信息科技有限公司 | 医学影像分割模型训练方法、分割方法及装置 |
CN113192077A (zh) * | 2021-04-15 | 2021-07-30 | 华中科技大学 | 一种细胞及区域层次的病理图自动分类方法及系统 |
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114972968A (zh) * | 2022-05-19 | 2022-08-30 | 长春市大众物流装配有限责任公司 | 基于多重神经网络的托盘识别和位姿估计方法 |
CN116012838A (zh) * | 2022-12-30 | 2023-04-25 | 创芯国际生物科技(广州)有限公司 | 基于人工智能的类器官活性识别方法及系统 |
CN116012838B (zh) * | 2022-12-30 | 2023-11-07 | 创芯国际生物科技(广州)有限公司 | 基于人工智能的类器官活性识别方法及系统 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109886986B (zh) | 一种基于多分支卷积神经网络的皮肤镜图像分割方法 | |
CN109410219B (zh) | 一种基于金字塔融合学习的图像分割方法、装置和计算机可读存储介质 | |
CN109886273B (zh) | 一种cmr图像分割分类系统 | |
US20220343623A1 (en) | Blood smear full-view intelligent analysis method, and blood cell segmentation model and recognition model construction method | |
CN109671094B (zh) | 一种基于频域分级的眼底图像血管分割方法 | |
CN109035172B (zh) | 一种基于深度学习的非局部均值超声图像去噪方法 | |
CN111369565B (zh) | 一种基于图卷积网络的数字病理图像的分割与分类方法 | |
CN111145209B (zh) | 一种医学图像分割方法、装置、设备及存储介质 | |
CN112308860A (zh) | 基于自监督学习的对地观测图像语义分割方法 | |
CN113920108A (zh) | 一种对用于处理细胞图像的U-Net模型进行训练的训练方法 | |
WO2008133951A2 (en) | Method and apparatus for image processing | |
JP2021528784A (ja) | パノラマ画像のスカイフィルタ方法及び携帯端末 | |
CN108305241B (zh) | 基于深度投票模型的sd-oct图像ga病变分割方法 | |
CN112446892A (zh) | 一种基于注意力学习的细胞核分割方法 | |
CN111161287A (zh) | 基于对称双向级联网络深度学习的视网膜血管分割方法 | |
CN112884788B (zh) | 基于丰富上下文网络的视杯视盘分割方法及成像方法 | |
CN113066025B (zh) | 一种基于增量学习与特征、注意力传递的图像去雾方法 | |
CN111179275A (zh) | 一种医学超声图像分割方法 | |
CN112581483B (zh) | 基于自学习的植物叶片叶脉分割方法和装置 | |
CN114972378A (zh) | 一种基于掩码注意力机制的脑肿瘤mri图像分割方法 | |
CN112927237A (zh) | 基于改进SCB-Unet网络的蜂窝肺病灶分割方法 | |
CN114881105A (zh) | 基于transformer模型和对比学习的睡眠分期方法及系统 | |
CN108961270B (zh) | 一种基于语义分割的桥梁裂缝图像分割模型 | |
CN113160146B (zh) | 一种基于图神经网络的变化检测方法 | |
CN112700409A (zh) | 视网膜微动脉瘤自动检测方法及成像方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |