CN112116009B - 基于卷积神经网络的新冠肺炎x射线图像识别方法及系统 - Google Patents
基于卷积神经网络的新冠肺炎x射线图像识别方法及系统 Download PDFInfo
- Publication number
- CN112116009B CN112116009B CN202010992932.4A CN202010992932A CN112116009B CN 112116009 B CN112116009 B CN 112116009B CN 202010992932 A CN202010992932 A CN 202010992932A CN 112116009 B CN112116009 B CN 112116009B
- Authority
- CN
- China
- Prior art keywords
- layer
- image recognition
- channel
- ray image
- ray
- 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.)
- Active
Links
- 238000013527 convolutional neural network Methods 0.000 title claims abstract description 29
- 238000000034 method Methods 0.000 title claims abstract description 25
- 206010035664 Pneumonia Diseases 0.000 title claims description 33
- 238000000605 extraction Methods 0.000 claims abstract description 64
- 238000012549 training Methods 0.000 claims description 39
- 238000011176 pooling Methods 0.000 claims description 30
- 208000025721 COVID-19 Diseases 0.000 claims description 19
- 238000012360 testing method Methods 0.000 claims description 18
- 238000010586 diagram Methods 0.000 claims description 15
- 238000007781 pre-processing Methods 0.000 claims description 13
- 238000004590 computer program Methods 0.000 claims description 11
- 230000002401 inhibitory effect Effects 0.000 claims description 8
- 230000009467 reduction Effects 0.000 claims description 6
- 238000012545 processing Methods 0.000 claims description 3
- 238000005549 size reduction Methods 0.000 claims description 3
- 230000002829 reductive effect Effects 0.000 abstract description 6
- 230000015556 catabolic process Effects 0.000 abstract description 3
- 238000006731 degradation reaction Methods 0.000 abstract description 3
- 238000004364 calculation method Methods 0.000 description 7
- 238000011976 chest X-ray Methods 0.000 description 4
- 238000012935 Averaging Methods 0.000 description 2
- 241000711573 Coronaviridae Species 0.000 description 2
- 238000013461 design Methods 0.000 description 2
- 238000012795 verification Methods 0.000 description 2
- 230000004913 activation Effects 0.000 description 1
- 230000003321 amplification Effects 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 239000002131 composite material Substances 0.000 description 1
- 238000002591 computed tomography Methods 0.000 description 1
- 230000008094 contradictory effect Effects 0.000 description 1
- 230000003247 decreasing effect Effects 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 230000000694 effects Effects 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 230000006870 function Effects 0.000 description 1
- 230000005764 inhibitory process Effects 0.000 description 1
- 238000010606 normalization Methods 0.000 description 1
- 238000003199 nucleic acid amplification method Methods 0.000 description 1
- 230000008569 process Effects 0.000 description 1
- 238000011946 reduction process Methods 0.000 description 1
- 230000035945 sensitivity Effects 0.000 description 1
- 238000010200 validation analysis Methods 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
-
- 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
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V2201/00—Indexing scheme relating to image or video recognition or understanding
- G06V2201/03—Recognition of patterns in medical or anatomical images
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Bioinformatics & Computational Biology (AREA)
- Computational Linguistics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Evolutionary Biology (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Image Analysis (AREA)
Abstract
本发明公开一种基于卷积神经网络的新冠肺炎X射线图像识别方法及系统,该方法针对X射线图像类别间相似度高且类别内变异性低的问题,构建了包括若干通道特征权重提取模块的图像识别模型,该通道特征权重提取模块可有效预防随着图像识别模型层数增多而退化的问题、减少模型的参数量,利用该通道特征权重提取模块可精确、快速地获取图像特征提取过程中每个通道的权重系数,放大权重靠前的通道特征、抑制权重靠后的通道特征,通过在模型中设计若干该通道特征权重提取模块可有效增强模型的特征提取能力,尤其是对X射线图像的特征提取能力,因此利用本发明提供的基于卷积神经网络的图像识别模型进行新冠肺炎X射线图像识别的精度高且识别速率快。
Description
技术领域
本发明涉及医学X-Ray图像识别技术领域,尤其是一种基于卷积神经网络的新冠肺炎X射线图像识别方法及系统。
背景技术
COVID-19为新型冠状病毒肺炎,简称新冠肺炎。COVID-19包含一些可以被胸部X射线(X-Ray)检测到的放射学特征,因此可通过对X-Ray图像进行识别筛选出含COVID-19放射学特征的X-Ray图像。但是现有技术中还没有一种可准确、快速地识别出含COVID-19放射学特征的X-Ray图像的方法。
发明内容
本发明提供一种基于卷积神经网络的新冠肺炎X射线图像识别方法及系统,用于克服现有技术中准确度不高、识别速率较低等缺陷。
为实现上述目的,本发明提出一种基于卷积神经网络的新冠肺炎X射线图像识别方法,包括:
获取若干带标识的X-Ray图像;若干所述X-Ray图像中包括含COVID-19放射学特征的图像、含普通肺炎放射学特征的图像和正常图像;
对X-Ray图像进行预处理,并将预处理后的X-Ray图像分成训练集和测试集;
利用训练集对预先构建的基于卷积神经网络的图像识别模型进行训练,利用测试集对训练后的图像识别模型进行测试;所述图像识别模型包括若干通道特征权重提取模块,所述通道特征权重提取模块用于获取图像特征提取过程中每个通道的权重系数,放大权重靠前的通道特征、抑制权重靠后的通道特征;
利用测试后的图像识别模型对待识别的X-Ray图像进行识别,获得X-Ray图像的类别。
为实现上述目的,本发明还提出一种基于卷积神经网络的新冠肺炎X射线图像识别系统,包括:
图像获取模块,用于获取若干带标识的X-Ray图像;若干所述X-Ray图像中包括含COVID-19放射学特征的图像、含普通肺炎放射学特征的图像和正常图像;
预处理模块,用于对X-Ray图像进行预处理,并将预处理后的X-Ray图像分成训练集和测试集;
模型训练模块,用于利用训练集对预先构建的基于卷积神经网络的图像识别模型进行训练,利用测试集对训练后的图像识别模型进行测试;所述图像识别模型包括若干通道特征权重提取模块,所述通道特征权重提取模块用于获取图像特征提取过程中每个通道的权重系数,放大权重靠前的通道特征、抑制权重靠后的通道特征;
图像识别模块,用于利用测试后的图像识别模型对待识别的X-Ray图像进行识别,获得X-Ray图像的类别。
为实现上述目的,本发明还提出一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时实现上述所述方法的步骤。
为实现上述目的,本发明还提出一种计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现上述所述的方法的步骤。
与现有技术相比,本发明的有益效果有:
本发明提供的基于卷积神经网络的新冠肺炎X射线图像识别方法,针对X-Ray图像类别间相似度高且类别内变异性低的问题,构建了包括若干通道特征权重提取模块的图像识别模型,该通道特征权重提取模块可有效预防随着图像识别模型层数增多而退化的问题、并减少图像识别模型的参数量,利用该通道特征权重提取模块可精确、快速地获取图像特征提取过程中每个通道的权重系数,放大权重靠前的通道特征、抑制权重靠后的通道特征,通过在图像识别模型中设计若干该通道特征权重提取模块可有效增强图像识别模型的特征提取能力,尤其是对X-Ray图像的特征提取能力,因此利用本发明提供的基于卷积神经网络的图像识别模型进行COVID-19 X-Ray图像识别的精度高且识别速率快。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图示出的结构获得其他的附图。
图1为本发明提供的基于卷积神经网络的新冠肺炎X射线图像识别方法流程图;
图2为COVID-19 X-Ray图像;
图3为普通肺炎X-Ray图像;
图4为正常X-Ray图像;
图5为本发明实施例中通道特征权重提取模块的结构图;
图6为本发明实施例中图像识别模型的分类器结构图;
图7为本发明另一实施例中图像识别模型的分类器结构图;
图8为不同图像识别模型的参数量对比图;
图9为不同图像识别模型的计算量对比图。
本发明目的的实现、功能特点及优点将结合实施例,参照附图做进一步说明。
具体实施方式
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明的一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
另外,本发明各个实施例之间的技术方案可以相互结合,但是必须是以本领域普通技术人员能够实现为基础,当技术方案的结合出现相互矛盾或无法实现时应当认为这种技术方案的结合不存在,也不在本发明要求的保护范围之内。
本发明提出一种基于卷积神经网络的新冠肺炎X射线图像识别方法,如图1所示,包括:
101:获取若干带标识的X-Ray图像;若干所述X-Ray图像中包括含COVID-19放射学特征的图像、含普通肺炎放射学特征的图像和正常图像;
102:对X-Ray图像进行预处理,并将预处理后的X-Ray图像分成训练集和测试集;
预处理以使处理后的图像可顺利输入图像识别模型中,并实现X-Ray图像的扩充。
103:利用训练集对预先构建的基于卷积神经网络的图像识别模型进行训练,利用测试集对训练后的图像识别模型进行测试;所述图像识别模型包括若干通道特征权重提取模块,所述通道特征权重提取模块用于获取图像特征提取过程中每个通道的权重系数,放大权重靠前的通道特征、抑制权重靠后的通道特征;
通道,指图像识别模型中输出特征的通道。
通道特征,指的是对应得这个通道的特征图。
104:利用测试后的图像识别模型对待识别的X-Ray图像进行识别,获得X-Ray图像的类别。
本发明提供的基于卷积神经网络的COVID-19 X-Ray图像识别方法,针对X-Ray图像类别间相似度高且类别内变异性低的问题,构建了包括若干通道特征权重提取模块的图像识别模型,该通道特征权重提取模块可有效预防随着图像识别模型层数增多而退化的问题(随着图像识别模型层数的增多,X-Ray图像的信息会产生丢失,从而导致图像识别模型识别性能退化)、并减少图像识别模型的参数量,利用该通道特征权重提取模块可精确、快速地获取图像特征提取过程中每个通道的权重系数,放大权重靠前的通道特征、抑制权重靠后的通道特征,通过在图像识别模型中设计若干该通道特征权重提取模块可有效增强图像识别模型的特征提取能力,尤其是对X-Ray图像的特征提取能力,因此利用本发明提供的基于卷积神经网络的图像识别模型进行COVID-19 X-Ray图像识别的精度高且识别速率快。
在其中一个实施例中,对于步骤101,采用两个开源数据集,第一个数据集来自GitHub(https://github.com/ieee8023/covid-chestxray-dataset),该数据集由感染了COVID-19和其他肺炎的不同患者的X射线和CT扫描图像组成,一共有760张图像,本实施例选取了其中412幅COVID-19阳性患者的X射线(X-Ray)图像。第二个数据集来自Kaggle的胸部X射线图像(肺炎)(https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia),这个数据集由正常和普通肺炎患者的胸部X射线图像组成,其中包含了5863张胸部X射线图像,本实施例从该数据集中选取了4265张普通肺炎X射线图像和1575张正常X射线图像。COVID-19 X-Ray图像、普通肺炎X-Ray图像和正常X-Ray图像分别如图2、图3和图4所示,从图可知,X-Ray图像类别间相似度高并且类别内变异性低,这增加了模型对X-Ray图像进行识别的难度。
在下一个实施例中,对于步骤102,对X-Ray图像进行预处理,包括:
对X-Ray图像进行缩放处理以便于输入至图像识别模型中;
对X-Ray图像进行旋转、放大、亮度调整、对比度调整和倾斜,以扩充X-Ray图像数量,降低模型过拟合风险。
在本实施例中,还包括将预处理后的X-Ray图像分成训练集和验证集,训练集包含5526张X-Ray图像,其中COVID-19患者图像310张,正常图像1341张,普通肺炎3875张。测试集包含726张X-Ray图像,其中COVID-19患者X射线图像102张,正常X射线图像234张,普通肺炎X射线图像390张。
在另一个实施例中,对于步骤103,由于X-Ray图像类别间相似度高并且类别内变异性低,这将导致图像识别模型偏差和过度拟合,从而导致图像识别模型的性能和泛化性降低。因此,本发明设计了通道特征权重提取模块。
通道特征权重提取模块,如图5所示,依次包括:
短连接层,包含多种尺寸的卷积核,用于减少图像识别模型的参数量;所述短连接层的输入端和输出端进行短连接,用于预防X-Ray图像信息丢失;
全局平均池化层(GAP),用于将通道上的特征图压缩成一个全局特征;所述池化层的输入为短连接层的输出特征图与输入特征图进行对应特征值的相加后的特征图;
第一全连接层(FC1),用于降低特征维度;
第二全连接层(FC2),用于恢复特征维度,输出获得每个通道的权重系数;
输出层,用于将每个通道与对应的权重系数相乘,放大权重靠前的通道特征、抑制权重靠后的通道特征。
通过通道特征权重提取模块的设计,图像识别模型可以学习到每个通道的权重系数。在特征提取过程中,权重系数可以帮助图像识别模型提取到更多重要的通道特征,抑制不重要的通道特征,增强图像识别模型的特征提取能力。
在某个实施例中,短连接层如图5所示,依次包括1×1的卷积层(Conv1)、3×3的卷积层(Conv3)和1×1的卷积层(Conv1);
第一个1×1的卷积层用于降低特征图维度,第二个1×1的卷积层用于升高特征图维度。
“Conv”表示为一个包含了“卷积”、“批标准化”和“激活函数”的复合结构。
在下一个实施例中,图像识别模型包括:
输入层,用于对输入的X-Ray图像进行初步特征提取,所述输入层包括1个卷积层和1个最大池化层;
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,所述短连接模块和通道特征权重提取模块交替连接;所述短连接模块包括若干串连的短连接层;
输出层,用于根据最后1个短连接输出的特征图进行图像识别,输出图像类别;所述输出层包括分类器(Classifier)和Softmax层。
在某个实施例中,构建了三个包含不同数量短连接层的图像识别模型CFW-Net56、CFW-Net107和CFW-Net158,具体如表1所示(图中每层后的标号为当前层的输出通道数量,如Conv1-64,64表示当前Conv1的输出通道数量为64):
CFW-Net56:输入层,1个7×7的卷积层(Conv7,64个输出通道,步长(stride)为2)和1个3×3的最大池化层(Maxpool,步长为2);
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,短连接模块和通道特征权重提取模块交替连接;第一个短连接模块包括2个串连的短连接层,第二个短连接模块包括3个串连的短连接层,第三个短连接模块包括5个串连的短连接层,第四个短连接模块包括3个串连的短连接层;
输出层,分类器(Classifier)和Softmax层。
CFW-Net107:输入层,1个7×7的卷积层(Conv7,64个输出通道,步长(stride)为2)和1个3×3的最大池化层(Maxpool,步长为2);
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,短连接模块和通道特征权重提取模块交替连接;第一个短连接模块包括2个串连的短连接层,第二个短连接模块包括3个串连的短连接层,第三个短连接模块包括22个串连的短连接层,第四个短连接模块包括3个串连的短连接层;
输出层,分类器(Classifier)和Softmax层。
CFW-Net158:输入层,1个7×7的卷积层(Conv7,64个输出通道,步长(stride)为2)和1个3×3的最大池化层(Maxpool,步长为2);
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,短连接模块和通道特征权重提取模块交替连接;第一个短连接模块包括2个串连的短连接层,第二个短连接模块包括7个串连的短连接层,第三个短连接模块包括35个串连的短连接层,第四个短连接模块包括3个串连的短连接层;
输出层,分类器(Classifier)和Softmax层。
表1三种图像识别模型的具体结构表
在下一个实施例中,基于传统的AlexNet,VGGNets等网络以三层全连接层作为分类器,其包含了大量参数,对内存要求极高,因此本实施例采用单层全连接层(FC)作为分类器。
在其中一个实施例中,分类器为GFC分类器,如图6所示,包括1个全局平均池化层(GAP)和1个全连接层(FC);
所述全局平均池化层用于将输入分类器的特征图的尺寸降为1×1;
所述全连接层用于根据降尺寸后的特征图进行图像识别。
因为卷积层输出的特征图数量极多,使用单层全连接层作为分类器仍然会导致参数量过大。因此本实施例通过先用全局平均池化层将特征图尺寸降为1×1,然后再使用全连接层进行分类,大大减少了参数量。
在另一个实施例中,分类器为CGAP分类器,如图7所示,包括1个1×1的点卷积层(Conv1)和1个全局平均池化层(GAP);
所述点卷积层用于对输入分类器的特征图进行降维处理;
所述全局平均池化层用于根据降维后的特征图进行图像识别。
本实施例根据GAP结构的特性,通过使用1×1的点卷积层将输入分类器的特征进行降维处理,然后再用GAP进行分类。这种分类器没有全连接层,进一步减少了参数量。
在某个实施例中,为了研究不同深度和不同分类器对图像识别模型计算量和参数量的影响,对比了分别使用不同分类器的图像识别模型和不同深度图像识别模型的参数量和计算量。以三分类任务为例,设图像识别模型最后一层输出特征图大小为H×W×D(H为高度,W为宽度,D为通道数)。当使用单层全连接层(FC)作为分类器时,分类器参数量为3×H×W×D+3。当使用GFC分类器时,以全局平均池化代替了全连接层。由于池化层没有参数,因此可以进一步减少参数量,节省内存,分类器的参数量为D+D×3+3。当使用CGAP分类器时,分类器的参数量为D×3+H×W×3+3。对比结果如图8和图9所示。由图8可知,相同深度下的图像识别模型,使用FC作为分类器的模型比使用GFC分类器和CGAP分类器的模型多了大约29万参数量。因此在保证准确率的情况下应该避免使用FC作为分类器,这样可以减少模型参数量。虽然使用不同分类器的模型的参数量有区别,但是相比模型深度,分类器对本发明图像识别模型参数量影响不大。CFW-Net158-CGAP的参数量是CFW-Net56-CGAP的2.46倍,CFW-Net107-CGAP的参数量是CFW-Net56-CGAP的1.8倍,由此可以看出模型深度对模型参数量影响最大。所以当设备内存不足,硬件条件不支持过多的参数量时,CFW-Net56是一个比较好的选择。由图9可知,计算量受模型深度的影响比较大,CFW-Net158的计算量是CFW-Net107的1.47倍,CFW-Net107的计算量是CFW-Net56的1.91倍,CFW-Net158和CFW-Net107的计算量十分庞大。因此当模型准确率差距不大时,CFW-Net56模型具有最高的性价比。
此外,通过对比使用三种不同分类器的模型的计算量和参数量,可以发现使用CGAP和GFC的模型参数量比使用FC的网络节省了大约29万参数量,使用GFC的模型的计算量比使用CGAP和FC的网络计算量节省了大约30万计算量。因此,在准确率得到保证的情况下,优先考虑使用GFC分类器。
在下一个实施例中,对于步骤103,在对图像识别模型进行训练和验证之前,先把训练集和测试集中的X-Ray图像下采样为224×224尺寸的固定分辨率,然后对训练集数据进行数据增强:以0.75的概率在-10°到10°之间随机旋转图像,以0.75的概率在1到1.1倍之间随机放大图片,以0.75的概率在0.4和0.6之间进行随机调整图片亮度,以0.75的概率在0.8和1.25之间进行随机调整图片对比度,以0.75的概率在-0.2到0.2的幅度之间倾斜图片。经过数据增强,训练集扩充了4倍,一定程度上解决了X-Ray图像不足的问题,减少了模型过拟合的风险。
在训练过程中,采用学习率预热训练的方式,即从一个较小的学习率逐渐变大。然后再使用学习率衰减方式逐渐减少学习率。通过反复实验,最终的训练参数为:初始学习率设置为0.00004,共训练20个周期。在0-10个epoch中,学习率从0.00004逐步增加到0.0001。在11-20个epoch中,学习率从0.0001逐步减少到0.000046。
在本实施例中,构建了9种不同深度和不同分类器的图像识别模型对X-Ray图像进行识别,结果如表2所示,由表2可知,使用FC作为分类器的CFW-Net的性能低于使用另外两种分类器的模型。CFW-Net158-CGAP在数据集上的整体表现最好,整体准确率,精度,灵敏度,特异度和F1(F1分数是精度和召回率的加权平均值)得分都最高,分别为95.18%,96.65%,94.56%,97.60%,95.60%,但其COVID-19分类准确率较低。CFW-Net56-GFC在COVID-19类别准确率达到了最高100%,其整体准确率为94.35,比CFW-Net107高了1%左右,比CFW-Net158低了1%左右,表明随着模型的不断加深,模型的性能没有明显变化。CFW-Net158和CFW-Net107的计算量分别是CFW-Net56的2.81倍和1.91倍,参数量分别是CFW-Net56的2.46倍和1.80倍。虽然CFW-Net158的识别性能最好,但是其计算量和参数量相比CFW-Net56分别增加了2.81倍和2.46倍,准确率却只高了不到1%。CFW-Net56-GFC在COVID-19类别上的准确率最高且整体准确率高于CFW-Net56-CGAP和CFW-Net56-FC。经过综合考量,CFW-Net56-GFC性价比最高。
表2 9种不同图像识别模型对X-Ray图像进行识别的结果表
本实施例进一步将CFW-Net56-GFC与传统卷积神经网络VGG-19,GoogLeNet,ResNet-50,DenseNet-121进行比较(以本发明的训练集作为测试样本),结果如表3所示,由表3可知,DenseNet121在训练集上达到了不错的分类准确率,但准确率仍然低于本发明的CFW-Net56-GFC,且其因为深度的增加导致计算量和参数量明显高于CFW-Net56-GFC。GoogleNet在训练集上分类准确率最低。Vgg19的准确率比CFW-Net56-GFC低了1%左右,因为其深度较浅,难以充分提取图像特征,导致图像份分类准确率较低。而且由于VGG19使用三层全连接层作为分类器,导致参数量和计算量十分庞大,对设备要求极高,计算时间较长。ResNet50在训练集上实现了不错的识别性能,但是其各项性能指标全部低于CFW-Net56-GFC。
表3采用不同模型进行X-Ray图像识别的结果表
综上,本发明提供的基于卷积神经网络的图像识别模型对COVID-19 X-Ray图像的识别准确率高、速率快,尤其对COVID-19 X-Ray图像的识别准确率很高。
本发明还提出一种基于卷积神经网络的新冠肺炎X射线图像识别系统,包括:
图像获取模块,用于获取若干带标识的X-Ray图像;若干所述X-Ray图像中包括含COVID-19放射学特征的图像、含普通肺炎放射学特征的图像和正常图像;
预处理模块,用于对X-Ray图像进行预处理,并将预处理后的X-Ray图像分成训练集和验证集;
模型训练模块,用于利用训练集对预先构建的基于卷积神经网络的图像识别模型进行训练,利用验证集对训练后的图像识别模型进行验证;所述图像识别模型包括若干通道特征权重提取模块,所述通道特征权重提取模块用于获取图像识别模型中每个通道的权重系数,提取权重靠前的通道特征、抑制权重靠后的通道特征;
图像识别模块,用于利用验证后的图像识别模型对待识别的X-Ray图像进行识别,获得X-Ray图像的类别。
在其中一个实施例中,预处理模块还包括:
对X-Ray图像进行缩放处理以便于输入至图像识别模型中;
对X-Ray图像进行旋转、放大、亮度调整、对比度调整和倾斜,以扩充X-Ray图像数量。
在下一个实施例中,对于模型训练模块,通道特征权重提取模块依次包括:
短连接层,包含多种尺寸的卷积核,用于减少图像识别模型的参数量;所述短连接层的输入端和输出端进行短连接,用于预防X-Ray图像信息丢失;
全局平均池化层,用于将通道上的特征图压缩成一个全局特征;所述池化层的输入为短连接层的输出特征图与输入特征图进行对应特征值的相加后的特征图;
第一全连接层,用于降低特征维度;
第二全连接层,用于恢复特征维度,输出获得每个通道的权重系数;
输出层,用于将每个通道与对应的权重系数相乘,放大权重靠前的通道特征、抑制权重靠后的通道特征。
在另一个实施例中,对于模型训练模块,所述短连接层依次包括1×1的卷积层、3×3的卷积层和1×1的卷积层;
第一个1×1的卷积层用于降低特征图维度,第二个1×1的卷积层用于升高特征图维度。
在某个实施例中,对于模型训练模块,图像识别模型包括:
输入层,用于对输入的X-Ray图像进行初步特征提取,所述输入层包括1个卷积层和1个最大池化层;
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,所述短连接模块和通道特征权重提取模块交替连接;所述短连接模块包括若干串连的短连接层;
输出层,用于根据最后1个短连接输出的特征图进行图像识别,输出图像类别;所述输出层包括分类器和Softmax层。
在下一个实施例中,对于模型训练模块,分类器为GFC分类器,包括1个全局平均池化层和1个全连接层;
所述全局平均池化层用于将输入分类器的特征图的尺寸降为1×1;
所述全连接层用于根据降尺寸后的特征图进行图像识别。
在另一个实施例中,对于模型训练模块,分类器为CGAP分类器,包括1个1×1的点卷积层和1个全局平均池化层;
所述点卷积层用于对输入分类器的特征图进行降维处理;
所述全局平均池化层用于根据降维后的特征图进行图像识别。
本发明还提出一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时实现上述所述方法的步骤。
本发明还提出一种计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现上述所述的方法的步骤。
以上所述仅为本发明的优选实施例,并非因此限制本发明的专利范围,凡是在本发明的发明构思下,利用本发明说明书及附图内容所作的等效结构变换,或直接/间接运用在其他相关的技术领域均包括在本发明的专利保护范围内。
Claims (8)
1.一种基于卷积神经网络的新冠肺炎X射线图像识别方法,其特征在于,包括:
获取若干带标识的X-Ray图像;若干所述X-Ray图像中包括含COVID-19放射学特征的图像、含普通肺炎放射学特征的图像和正常图像;
对X-Ray图像进行预处理,并将预处理后的X-Ray图像分成训练集和测试集;
利用训练集对预先构建的基于卷积神经网络的图像识别模型进行训练,利用测试集对训练后的图像识别模型进行测试;所述图像识别模型包括:
输入层,用于对输入的X-Ray图像进行初步特征提取,所述输入层包括1个卷积层和1个最大池化层;
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,所述短连接模块和通道特征权重提取模块交替连接;所述短连接模块包括若干串连的短连接层;
输出层,用于根据最后1个短连接输出的特征图进行图像识别,输出图像类别;所述输出层包括分类器和Softmax层;
所述通道特征权重提取模块依次包括:
短连接层,包含多种尺寸的卷积核,用于减少图像识别模型的参数量;所述短连接层的输入端和输出端进行短连接,用于预防X-Ray图像信息丢失;
全局平均池化层,用于将通道上的特征图压缩成一个全局特征;所述池化层的输入为短连接层的输出特征图与输入特征图进行对应特征值的相加后的特征图;
第一全连接层,用于降低特征维度;
第二全连接层,用于恢复特征维度,输出获得每个通道的权重系数;
输出层,用于将每个通道与对应的权重系数相乘,放大权重靠前的通道特征、抑制权重靠后的通道特征;
利用测试后的图像识别模型对待识别的X-Ray图像进行识别,获得X-Ray图像的类别。
2.如权利要求1所述的基于卷积神经网络的新冠肺炎X射线图像识别方法,其特征在于,对X-Ray图像进行预处理,包括:
对X-Ray图像进行缩放处理以便于输入至图像识别模型中;
对X-Ray图像进行旋转、放大、亮度调整、对比度调整和倾斜,以扩充X-Ray图像数量。
3.如权利要求2所述的基于卷积神经网络的新冠肺炎X射线图像识别方法,其特征在于,所述短连接层依次包括1×1的卷积层、3×3的卷积层和1×1的卷积层;
第一个1×1的卷积层用于降低特征图维度,第二个1×1的卷积层用于升高特征图维度。
4.如权利要求3所述的基于卷积神经网络的新冠肺炎X射线图像识别方法,其特征在于,所述分类器为GFC分类器,包括1个全局平均池化层和1个全连接层;
所述全局平均池化层用于将输入分类器的特征图的尺寸降为1×1;
所述全连接层用于根据降尺寸后的特征图进行图像识别。
5.如权利要求3所述的基于卷积神经网络的新冠肺炎X射线图像识别方法,其特征在于,所述分类器为CGAP分类器,包括1个1×1的点卷积层和1个全局平均池化层;
所述点卷积层用于对输入分类器的特征图进行降维处理;
所述全局平均池化层用于根据降维后的特征图进行图像识别。
6.一种基于卷积神经网络的新冠肺炎X射线图像识别系统,其特征在于,包括:
图像获取模块,用于获取若干带标识的X-Ray图像;若干所述X-Ray图像中包括含COVID-19放射学特征的图像、含普通肺炎放射学特征的图像和正常图像;
预处理模块,用于对X-Ray图像进行预处理,并将预处理后的X-Ray图像分成训练集和测试集;
模型训练模块,用于利用训练集对预先构建的基于卷积神经网络的图像识别模型进行训练,利用测试集对训练后的图像识别模型进行测试;所述图像识别模型包括:
输入层,用于对输入的X-Ray图像进行初步特征提取,所述输入层包括1个卷积层和1个最大池化层;
特征提取子网络,包括4个短连接模块和3个通道特征权重提取模块,所述短连接模块和通道特征权重提取模块交替连接;所述短连接模块包括若干串连的短连接层;
输出层,用于根据最后1个短连接输出的特征图进行图像识别,输出图像类别;所述输出层包括分类器和Softmax层;
所述通道特征权重提取模块依次包括:
短连接层,包含多种尺寸的卷积核,用于减少图像识别模型的参数量;所述短连接层的输入端和输出端进行短连接,用于预防X-Ray图像信息丢失;
全局平均池化层,用于将通道上的特征图压缩成一个全局特征;所述池化层的输入为短连接层的输出特征图与输入特征图进行对应特征值的相加后的特征图;
第一全连接层,用于降低特征维度;
第二全连接层,用于恢复特征维度,输出获得每个通道的权重系数;
输出层,用于将每个通道与对应的权重系数相乘,放大权重靠前的通道特征、抑制权重靠后的通道特征;
图像识别模块,用于利用测试后的图像识别模型对待识别的X-Ray图像进行识别,获得X-Ray图像的类别。
7.一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现权利要求1~5中任一项所述方法的步骤。
8.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1~5中任一项所述的方法的步骤。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010992932.4A CN112116009B (zh) | 2020-09-21 | 2020-09-21 | 基于卷积神经网络的新冠肺炎x射线图像识别方法及系统 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010992932.4A CN112116009B (zh) | 2020-09-21 | 2020-09-21 | 基于卷积神经网络的新冠肺炎x射线图像识别方法及系统 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112116009A CN112116009A (zh) | 2020-12-22 |
CN112116009B true CN112116009B (zh) | 2024-04-26 |
Family
ID=73800207
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010992932.4A Active CN112116009B (zh) | 2020-09-21 | 2020-09-21 | 基于卷积神经网络的新冠肺炎x射线图像识别方法及系统 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112116009B (zh) |
Families Citing this family (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112784856A (zh) * | 2021-01-29 | 2021-05-11 | 长沙理工大学 | 胸部x射线图像的通道注意力特征提取方法和识别方法 |
CN113111835B (zh) * | 2021-04-23 | 2022-08-02 | 长沙理工大学 | 卫星遥感图像语义分割方法、装置、电子设备和存储介质 |
CN113129293B (zh) * | 2021-04-26 | 2022-08-23 | 长沙理工大学 | 医学图像分类方法、装置、计算机设备和存储介质 |
CN113409300A (zh) * | 2021-07-12 | 2021-09-17 | 上海市第一人民医院 | 基于人工智能技术的新冠肺炎数据处理系统 |
CN113593714A (zh) * | 2021-07-26 | 2021-11-02 | 陕西师范大学 | 一种多分类新冠肺炎病例的检测方法、系统、设备及介质 |
Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN103473558A (zh) * | 2013-09-04 | 2013-12-25 | 深圳先进技术研究院 | 基于神经网络的图像识别方法和系统 |
CN110427922A (zh) * | 2019-09-03 | 2019-11-08 | 陈�峰 | 一种基于机器视觉和卷积神经网络病虫害识别系统和方法 |
CN110866907A (zh) * | 2019-11-12 | 2020-03-06 | 中原工学院 | 基于注意力机制的全卷积网络织物疵点检测方法 |
CN111046964A (zh) * | 2019-12-18 | 2020-04-21 | 电子科技大学 | 一种基于卷积神经网络的人和车辆红外热图像识别方法 |
CN111259982A (zh) * | 2020-02-13 | 2020-06-09 | 苏州大学 | 一种基于注意力机制的早产儿视网膜图像分类方法和装置 |
CN111325739A (zh) * | 2020-02-28 | 2020-06-23 | 北京推想科技有限公司 | 肺部病灶检测的方法及装置,和图像检测模型的训练方法 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
KR101836096B1 (ko) * | 2016-12-02 | 2018-03-12 | 주식회사 수아랩 | 이미지 데이터의 상태 판단 방법, 장치 및 컴퓨터-판독가능 저장 매체에 저장된 컴퓨터 프로그램 |
-
2020
- 2020-09-21 CN CN202010992932.4A patent/CN112116009B/zh active Active
Patent Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN103473558A (zh) * | 2013-09-04 | 2013-12-25 | 深圳先进技术研究院 | 基于神经网络的图像识别方法和系统 |
CN110427922A (zh) * | 2019-09-03 | 2019-11-08 | 陈�峰 | 一种基于机器视觉和卷积神经网络病虫害识别系统和方法 |
CN110866907A (zh) * | 2019-11-12 | 2020-03-06 | 中原工学院 | 基于注意力机制的全卷积网络织物疵点检测方法 |
CN111046964A (zh) * | 2019-12-18 | 2020-04-21 | 电子科技大学 | 一种基于卷积神经网络的人和车辆红外热图像识别方法 |
CN111259982A (zh) * | 2020-02-13 | 2020-06-09 | 苏州大学 | 一种基于注意力机制的早产儿视网膜图像分类方法和装置 |
CN111325739A (zh) * | 2020-02-28 | 2020-06-23 | 北京推想科技有限公司 | 肺部病灶检测的方法及装置,和图像检测模型的训练方法 |
Non-Patent Citations (4)
Title |
---|
A deep learning approach to detect Covid-19 coronavirus with X-Ray images;Jain, Govardhan等;《Biocybernetics and biomedical engineering》;20200907;第40卷(第4期);1394-1397, 1402 * |
On Building Detection Using the Class Activation Map: Case Study on a Landsat8 Image;P Charuchinda等;《2019 10th International Conference of Information and Communication Technology for Embedded Systems (IC-ICTES)》;20190425;2-3 * |
基于深度监督显著目标检测的草莓图像分割;钱文秀等;《华东理工大学学报(自然科学版)》;20200228;第46卷(第01期);114-120 * |
基于结构特征增强的图像显著性检测方法研究;赵婷;《中国优秀硕士学位论文全文数据库 (信息科技辑)》;20200215(第02期);I138-1659 * |
Also Published As
Publication number | Publication date |
---|---|
CN112116009A (zh) | 2020-12-22 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112116009B (zh) | 基于卷积神经网络的新冠肺炎x射线图像识别方法及系统 | |
CN108596258B (zh) | 一种基于卷积神经网络随机池化的图像分类方法 | |
CN108038476B (zh) | 一种基于边缘检测与sift的人脸表情识别特征提取方法 | |
CN112560968B (zh) | 一种基于卷积和残差网络的her2图像分类方法及系统 | |
CN109493342B (zh) | 一种基于深度学习的皮肤病图片病变类型分类方法 | |
CN110751636B (zh) | 一种基于改进型编解码网络的眼底图像视网膜动脉硬化检测方法 | |
Zhang et al. | Diabetic retinopathy classification using deeply supervised ResNet | |
CN112784856A (zh) | 胸部x射线图像的通道注意力特征提取方法和识别方法 | |
JP6945253B2 (ja) | 分類装置、分類方法、プログラム、ならびに、情報記録媒体 | |
CN112767406B (zh) | 角膜溃疡分割的深度卷积神经网络训练方法及分割方法 | |
CN113011253B (zh) | 基于ResNeXt网络的人脸表情识别方法、装置、设备及存储介质 | |
CN113012163A (zh) | 一种基于多尺度注意力网络的视网膜血管分割方法、设备及存储介质 | |
Jin et al. | White matter hyperintensity segmentation from T1 and FLAIR images using fully convolutional neural networks enhanced with residual connections | |
CN112686336A (zh) | 一种基于神经网络的烧伤创面深度分类系统 | |
CN113420793A (zh) | 一种基于改进的卷积神经网络ResNeSt50的胃印戒细胞癌分类方法 | |
CN117409002A (zh) | 一种用于创伤的视觉识别检测系统及其检测方法 | |
US11538158B2 (en) | Convolutional neural network and associated method for identifying basal cell carcinoma | |
CN114049935A (zh) | 一种基于多卷积神经网络的her2图像分类系统 | |
CN111598144B (zh) | 图像识别模型的训练方法和装置 | |
CN113506274A (zh) | 基于视觉显著性差异图的用于人体认知状况的检测系统 | |
CN114418999B (zh) | 基于病变关注金字塔卷积神经网络的视网膜病变检测系统 | |
CN110765908A (zh) | 基于深度学习的级联式癌细胞检测系统 | |
CN112862089B (zh) | 一种具有可解释性的医学图像深度学习方法 | |
CN115830686A (zh) | 基于特征融合的生物识别方法、系统、装置及存储介质 | |
Jain et al. | Brain Tumor Detection using MLops and Hybrid Multi-Cloud |
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 | ||
GR01 | Patent grant | ||
GR01 | Patent grant |