CN115578590A - 基于卷积神经网络模型的图像识别方法、装置及终端设备 - Google Patents
基于卷积神经网络模型的图像识别方法、装置及终端设备 Download PDFInfo
- Publication number
- CN115578590A CN115578590A CN202211254852.4A CN202211254852A CN115578590A CN 115578590 A CN115578590 A CN 115578590A CN 202211254852 A CN202211254852 A CN 202211254852A CN 115578590 A CN115578590 A CN 115578590A
- Authority
- CN
- China
- Prior art keywords
- feature map
- image
- neural network
- convolutional neural
- local feature
- 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
- 238000013527 convolutional neural network Methods 0.000 title claims abstract description 153
- 238000000034 method Methods 0.000 title claims abstract description 72
- 239000011159 matrix material Substances 0.000 claims abstract description 106
- 238000000605 extraction Methods 0.000 claims abstract description 82
- 238000012549 training Methods 0.000 claims description 33
- 238000012545 processing Methods 0.000 claims description 30
- 238000004590 computer program Methods 0.000 claims description 27
- 230000000694 effects Effects 0.000 abstract description 6
- 238000001514 detection method Methods 0.000 description 22
- 230000006870 function Effects 0.000 description 16
- 238000010586 diagram Methods 0.000 description 13
- 230000008569 process Effects 0.000 description 13
- 230000004913 activation Effects 0.000 description 6
- 238000013528 artificial neural network Methods 0.000 description 5
- 238000004364 calculation method Methods 0.000 description 5
- 125000004122 cyclic group Chemical group 0.000 description 4
- 239000000284 extract Substances 0.000 description 4
- 230000008878 coupling Effects 0.000 description 3
- 238000010168 coupling process Methods 0.000 description 3
- 238000005859 coupling reaction Methods 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 238000011176 pooling Methods 0.000 description 3
- 238000004891 communication Methods 0.000 description 2
- 238000002372 labelling Methods 0.000 description 2
- 238000003062 neural network model Methods 0.000 description 2
- 210000002569 neuron Anatomy 0.000 description 2
- 238000010606 normalization Methods 0.000 description 2
- 238000011160 research Methods 0.000 description 2
- 238000004458 analytical method Methods 0.000 description 1
- 230000008033 biological extinction Effects 0.000 description 1
- 238000004422 calculation algorithm Methods 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 238000004880 explosion Methods 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 230000003993 interaction Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 230000000737 periodic effect Effects 0.000 description 1
- 230000003068 static effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 238000012360 testing method Methods 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/764—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
-
- 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
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/40—Extraction of image or video features
- G06V10/42—Global feature extraction by analysis of the whole pattern, e.g. using frequency domain transformations or autocorrelation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
- G06V10/774—Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Computing Systems (AREA)
- General Health & Medical Sciences (AREA)
- Software Systems (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Artificial Intelligence (AREA)
- Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Databases & Information Systems (AREA)
- Medical Informatics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Molecular Biology (AREA)
- General Engineering & Computer Science (AREA)
- Mathematical Physics (AREA)
- Image Analysis (AREA)
- Image Processing (AREA)
Abstract
本申请适用于图像识别技术领域,提供了基于卷积神经网络模型的图像识别方法、装置及终端设备,其中,上述卷积神经网络基于待识别图像的权重矩阵对上述待识别图像进行全局特征提取,上述图像识别方法包括:将待识别图像输入到经训练的上述卷积神经网络模型,通过上述卷积神经网络模型对上述待识别图像依次进行特征提取和识别,得到识别结果。本申请可以提高卷积神经网络对不同尺寸图像的图像识别效果。
Description
技术领域
本申请属于图像识别技术领域,尤其涉及基于卷积神经网络模型的图像识别方法、装置、终端设备以及计算机可读存储介质。
背景技术
卷积神经网络是一种特殊的人工神经网络,它已经成为语音分析和图像识别领域最为常用的一种技术,因此也是人工神经网络研究热点。在卷积神经网络通常是一个多层的神经网络,每层由多个特征图(featureMap)组成,而每个特征图由多个独立神经元组成,同一特征图的神经元共享权重(即卷积核),卷积神经网络正是通过共享权重减少网络各层的连接,同时又降低了过拟合的风险。
现有的卷积神经网络在进行特征提取时通常采用一个小的核矩阵(即权重矩阵)和大特征图作卷积操作,以节约参数,但其共用一个静态的权重矩阵,不能很好地适应不同尺寸的输入图像。
发明内容
本申请实施例提供了基于卷积神经网络模型的图像识别方法、装置及终端设备,可以提高卷积神经网络模型对不同尺寸图像的图像识别效果。
第一方面,本申请实施例提供了一种基于卷积神经网络模型的图像识别方法,上述卷积神经网络模型基于待识别图像的权重矩阵对所述待识别图像进行全局特征提取,上述图像识别方法包括:
将待识别图像输入到经训练的所述卷积神经网络模型,通过所述卷积神经网络模型对所述待识别图像依次进行特征提取和识别,得到识别结果。
第二方面,本申请实施例提供了一种卷积神经网络模型训练方法,包括:
获取构建的卷积神经网络模型,并将样本图像输入到上述卷积神经网络进行训练,直至上述卷积神经网络满足预设要求,得到卷积神经网络模型;
其中,上述卷积神经网络基于样本图像的权重矩阵对上述样本图像进行全局特征提取。
第三方面,本申请实施例提供了一种图像识别装置,包括:
输入模块和经训练的卷积神经网络模型,所述卷积神经网络模型基于待识别图像的权重矩阵对所述待识别图像进行全局特征提取;
所述输入模块,用于将待识别图像输入到所述卷积神经网络模型;
所述卷积神经网络模型,用于对所述待识别图像依次进行特征提取和识别,得到识别结果。
第四方面,本申请实施例提供了一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现上述第一方面所述的基于卷积神经网络模型的图像识别方法或上述第二方面所述的卷积神经网络模型训练方法的步骤。
第五方面,本申请实施例提供了一种计算机可读存储介质,所述计算机存储介质存储有计算机程序,所述计算机程序被处理器执行时实现上述第一方面中所述的基于卷积神经网络模型的图像识别方法或上述第二方面所述的卷积神经网络模型训练方法的步骤。
第六方面,本申请实施例提供了一种计算机程序产品,当计算机程序产品在终端设备上运行时,使得终端设备执行上述第一方面中任一项所述的基于卷积神经网络模型的图像识别方法或上述第二方面所述的卷积神经网络模型训练方法。
本申请实施例与现有技术相比存在的有益效果是:
本申请实施例中,将待识别图像输入到经训练的所述卷积神经网络模型,通过所述卷积神经网络模型对所述待识别图像依次进行特征提取和识别,得到识别结果。由于该卷积神经网络模型基于待识别图像的权重矩阵对所述待识别图像进行全局特征提取,因此,在对不同尺寸的待识别图像进行图像识别时,可以根据各个待识别图像动态调整权重矩阵来对该待识别图像进行全局特征的提取,使得卷积神经网络模型对不同尺寸的输入图像具有良好的识别效果。
附图说明
为了更清楚地说明本申请实施例中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍。
图1是本申请一实施例提供的一种基于卷积神经网络模型的图像识别方法的流程示意图;
图2是本申请实施例提供的卷积神经网络模型的结构示意图;
图3是本申请实施例提供的用于动态提取权重矩阵的卷积模块的结构示意图;
图4是本申请实施例提供的第二卷积模块的结构示意图;
图5是本申请实施例提供的卷积神经网络模型训练方法的流程示意图;
图6是本申请实施例提供的基于卷积神经网络模型的图像识别装置的结构示意图;
图7是本申请实施例提供的卷积神经网络模型训练装置的结构示意图;
图8是本申请实施例提供的终端设备的结构示意图。
具体实施方式
以下描述中,为了说明而不是为了限定,提出了诸如特定系统结构、技术之类的具体细节,以便透彻理解本申请实施例。然而,本领域的技术人员应当清楚,在没有这些具体细节的其它实施例中也可以实现本申请。在其它情况中,省略对众所周知的系统、装置、电路以及方法的详细说明,以免不必要的细节妨碍本申请的描述。
应当理解,当在本申请说明书和所附权利要求书中使用时,术语“包括”指示所描述特征、整体、步骤、操作、元素和/或组件的存在,但并不排除一个或多个其它特征、整体、步骤、操作、元素、组件和/或其集合的存在或添加。
还应当理解,在本申请说明书和所附权利要求书中使用的术语“和/或”是指相关联列出的项中的一个或多个的任何组合以及所有可能组合,并且包括这些组合。
另外,在本申请说明书和所附权利要求书的描述中,术语“第一”、“第二”、“第三”等仅用于区分描述,而不能理解为指示或暗示相对重要性。
在本申请说明书中描述的参考“一个实施例”或“一些实施例”等意味着在本申请的一个或多个实施例中包括结合该实施例描述的特定特征、结构或特点。由此,在本说明书中的不同之处出现的语句“在一个实施例中”、“在一些实施例中”、“在其他一些实施例中”、“在另外一些实施例中”等不是必然都参考相同的实施例,而是意味着“一个或多个但不是所有的实施例”,除非是以其他方式另外特别强调。
实施例一:
图1示出了本发明实施例提供的一种基于卷积神经网络模型的图像识别方法的流程示意图,详述如下:
将待识别图像输入到经训练的卷积神经网络模型,通过上述卷积神经网络模型对上述待识别图像依次进行特征提取和识别,得到识别结果。
其中,上述卷积神经网络模型基于待识别图像的权重矩阵对该待识别图像进行全局特征提取,以进行图像识别,即,根据当前输入的待识别图像的权重矩阵对该待识别图像进行全局特征的提取。
上述权重矩阵是指卷积神经网络的卷积核。在图像处理时,给定一个输入图像中一个小区域中像素加权平均后成为输出图像中的每个对应像素,其中,权值由一个函数定义,该函数即卷积核,对于不同的输入图像,卷积神经网络的权重矩阵(卷积核)是固定不变的。
具体地,由于在进行图像识别任务时,待识别图像的大小不一,采用同一个固定的权重矩阵,不能随着输入图像动态调整权重矩阵,并且,传统卷积操作仅拥有局部感受野,不能很好地提取图像的全局特征,影响图像识别的准确度,因此,将待识别图像输入到上述卷积神经网络模型进行特征提取的过程中,根据该待识别图像的权重矩阵来对该待识别图像进行全局特征的提取,以根据得到的特征进行识别,得到图像识别结果,使上述卷积神经网络模型同时具有全局感受野和动态权重,从而提高卷积神经网络模型的识别准确度。例如,对待识别图像A,根据待识别图像A的权重矩阵a对待识别图像A进行全局特征的提取,并根据得到的全局特征进行识别,得到待识别图像A的识别结果,对待识别图像B进行识别时,则根据待识别图像B的权重矩阵b来对待识别图像B进行全局特征的提取和识别。
本申请实施例中,将待识别图像输入到经训练的卷积神经网络模型,通过上述卷积神经网络模型对上述待识别图像依次进行特征提取和识别,得到识别结果。由于上述卷积神经网络模型在特征提取的过程中,根据待识别图像的权重矩阵对该待识别图像进行全局特征的提取,卷积神经网络模型同时具有全局感受野和动态权重,因此,可以较好地对不同大小的待识别图像进行特征提取,从而提高卷积神经网络模型的识别准确度。
在一些实施例中,上述基于卷积神经网络模型的图像识别方法还包括:
获取待识别图像。
可选地,上述待识别图像可以是摄像设备采集得到的一个图像,也可以是摄像设备采集得到的视频流中的一个图像帧。
可选地,由于不同图像识别任务所需的待识别图像可能不同,所采用的摄像设备、采集待检测图像的规则等也不尽相同,因此,根据各应用领域相应的采集方法和采集规则等,获取相应的待识别图像。例如,对于行人重识别任务,需要获取安装的多个摄像头所采集的图像作为待识别图像,以进行行人重识别任务。
本申请实施例中,根据各应用领域中图像识别任务所需的图像,采用相应的采集方法和采集规则,获取符合图像识别任务要求的待识别图像,以便进行图像识别任务。
在一些实施例中,上述卷积神经网络模型包括特征提取模块和识别模块,上述步骤在通过上述卷积神经网络模型对上述待识别图像依次进行特征提取和识别,得到识别结果,包括:
A1、通过上述特征提取模块对上述待识别图像进行特征提取;
A2、基于上述识别模块对提取到的特征进行识别,得到识别结果。
可选地,由于图像识别包括图像分类、目标检测等不同的任务(如人脸识别、行人检测等),不同的图像识别任务在进行图像识别时,对同一特征所采用的识别方法不尽相同,因此,通过上述特征提取模块对输入的待识别图像进行特征提取,并将提取到的特征作为上述识别模块的输入,根据图像识别任务进行相应的识别,得到识别结果。上述识别模块中可包括一个或多个识别单元,不同的识别单元所进行的识别任务不同,例如,可以包括人脸识别单元和目标检测单元,通过人脸识别单元对提取到的特征进行人脸识别任务,或将提取到的特征分布输入人脸识别单元和目标检测单元,进行人脸识别和目标检测任务。
本申请实施例中,通过特征提取模块对待识别图像进行特征提取,并通过识别模块获取提取到的特征进行相应的识别,得到相应的识别结果,以提高各图像识别任务的识别效率。
在一些实施例中,上述特征提取模块包括第一卷积模块和第二卷积模块,步骤A1包括:
A11、基于上述第一卷积模块对上述待识别图像进行局部特征提取,得到局部特征图;
A12、基于上述第二卷积模块根据上述局部特征图的权重矩阵对上述局部特征图进行全局特征提取,得到全局特征图。
可选地,上述卷积神经网络模型可以基于现有的卷积神经网络进行构建,将浅层的卷积层作为第一卷积模块,并将深层的卷积层或自注意力替换为第二卷积模块,上述第一卷积模块采用普通卷积方式对待识别图像进行卷积处理,提取待识别图像的局部特征,输出待识别图像的局部特征图,并将该局部特征图作为第二卷积模块的输入,上述第二模块通过动态卷积的方式,获取局部特征图的权重矩阵并根据该权重矩阵对局部特征图进行全局特征的提取,得到全局特征图。例如,如图2所示的卷积神经网络模型,前两层为普通卷积的第一卷积模块,较深层的四个卷积层为第二卷积模块,第二卷积模块与识别模块相连,将待识别图像作为第一卷积模块的输入进行局部特征提取,并将输出的特征输入到第二卷积模块进行全局特征的提取,上述第二卷积模块输出的全局特征图作为识别模块的输入进行识别,从而输出相应的识别结果。
需要说明的是,上述卷积神经网络模型中第一卷积模块和第二卷积模块也可以采用交叉出现的结构,即第一卷积模块与第二卷积模块连接,该第二卷积模块的输出与另一个第一卷积模块连接的结构,也可以采用图2所示结构多次堆叠的结构,使特征提取模块输出的特征图为第二卷积模块所提取的全局特征图即可(即,使识别模块基于第二卷积模块提取的全局特征进行识别),并不该限定卷积神经网络模型中第一卷积模块(普通卷积层)与本申请实施例提供的第二卷积模块的具体结构。
本申请实施例中,由于通过卷积神经网络模型的第一卷积模块提取待识别图像的局部特征,并将得到的局部特征图作为第二卷积模块的输入,根据上述局部特征图的权重矩阵对上述局部特征图进行全局特征提取,因此,得到的全局特征同时包含局部特征和全局特征,提高了卷积神经网络模型的识别准确度。
在一些实施例中,上述第二卷积模块包括第一分支和第二分支,上述步骤A12包括:
将上述局部特征图沿通道方向拆分,得到第一局部特征图和第二局部特征图,并将上述第一局部特征图和上述第二局部特征图分别输入到上述第一分支和上述第二分支。
可选地,在第二卷积模块对输入的局部特征图进行全局特征提取时,先将局部特征图沿通道方向进行拆分(如沿通道方向均匀拆分为两份),得到第一局部特征图和第二局部特征图,在将第一局部特征图输入到第一分支,将第二局部特征图输入到第二分支,以分别对第一局部特征图和第二局部特征图进行全局特征的提取。
上述第一分支和上述第二分支分别根据输入的局部特征图的权重矩阵对相应的局部特征图进行特征提取,得到第一全局特征图和第二全局特征图。
可选地,第一分支根据第一局部特征图生成上述第一局部特征图的权重矩阵,并根据上述权重矩阵对上述第一局部特征图进行全局特征的提取,得到第一全局特征图;第二分支根据第二局部特征图生成上述第二局部特征图的权重矩阵,根据上述权重矩阵对上述第二局部特征图进行全局特征的提取,得到第二全局特征图。
将上述第一全局特征图和上述第二全局特征图沿通道方向进行拼接,得到全局特征图。
可选地,由于第一局部特征图和第二局部特征图是经局部特征图沿通道方向拆分后得到的,因此,在第一分支和第二分支分别提取得到第一局部特征图和第二局部特征图的特征后,将得到的第一全局特征图和第二全局特征图沿通道方向进行拼接,得到完整的全局特征图,以根据输入的待识别图像的完整的全局特征图进行后续处理。
本申请实施例中,由于将局部特征图沿通道方向拆分成两份输入到第一分支和第二分支,局部特征图的通道数减半,因此,基于通道数减半的局部特征图进行全局特征的提取,降低了全局特征提取的计算复杂度,从而降低对设备计算能力的要求,便于在算力较低的设备上部署应用。
在一些实施例中,第一分支和第二分支根据输入的局部特征图进行全局特征的提取时,包括:
根据上述第一局部特征图生成列维度的权重矩阵,得到第一权重矩阵,并根据上述第一权重矩阵对上述第一局部特征图沿列维度进行卷积处理,得到第一全局特征图。
具体地,在第一分支对第一局部特征图进行特征提取时,根据上述第一局部特征图沿列维度(即H维度)进行卷积处理,动态生成上述第一局部特征图的权重矩阵,得到第一权重矩阵,再根据上述第一权重矩阵对上述第一局部特征图进行全局特征提取,得到第一全局特征图。
可选地,为了减小卷积神经网络模型的计算复杂度,在根据上述第一局部特征图动态生成列维度的第一权重矩阵时,对上述第一局部特征图沿列维度进行一维卷积处理,即在高度方向上对每一行像素进行卷积操作,例如,第一局部特征图表示为(B,C,H,W),对上述第一局部特征图采用轻量级的一维卷积模块沿列维度进行卷积处理,生成上述第一局部特征图的第一权重矩阵(B,C,H,1)。由于第一权重矩阵是在列维度上对每一行像素进行卷积操作得到的,该第一权重矩阵具有全局感受野,因此,在得到上述第一局部特征图的第一权重矩阵后,根据上述第一权重矩阵对上述第一局部特征图沿列维度进行循环卷积,提取上述第一局部特征图的全局特征,得到第一全局特征图。例如,假设第一局部特征图的高度为4,批处理大小为1,由于对上述第一特征图沿列维度进行卷积操作得到第一权重矩阵,因此,将第一局部特征图表示为x=(x0,x1,x2,x3),第一权重矩阵(相当于卷积核)表示为w=(w0,w1,w2,w3),即相当于将第一局部特征图的每一行像素当作一个元素进行处理,使第一局部特征图和第一权重矩阵具有相同的分辨率(如尺寸为(C,H,1))。
可选地,在对上述第一局部特征图进行一维卷积处理以获取第一权重矩阵时,采用深度可分离卷积进行处理,以减少参数量和计算量,例如,如图3所示,一维卷积模块采用两层卷积层的轻量级网络结构,对第一局部特征图(B,C,H,W)沿列维度进行平均池化,使第一局部特征图降维,得到(B,C,H,1)形状的第一局部特征图,再对上述第一局部特征图采用1×3卷积核的深度可分离卷积,并采用HardSwish激活函数和批量归一化处理,再进行1×3深度可分离卷积,输出第一权重矩阵,其中,HardSwish激活函数是对Swish非线性激活函数的改进,Swish激活函数在一定程度上可以提高卷积神经网络的准确性,但其计算成本较高,不适合在嵌入式移动设备上使用,HardSwish激活函数可以实现为分段功能以减少内存访问次数,在提高卷积神经网络准确性的同时便于在嵌入式移动设备上部署使用。
根据上述第二分支的局部特征图生成行维度的权重矩阵,得到第二权重矩阵,并根据上述第二权重矩阵对上述第二局部特征图沿行维度进行卷积处理,得到第二全局特征图。
可选地,在第二分支对第二局部特征图进行特征提取时,根据上述第二局部特征图动态生成上述第二局部特征图的权重矩阵,得到第二权重矩阵,再根据上述第二权重矩阵和第二局部特征图进行卷积处理,得到第二全局特征图。
可选地,在根据上述第二局部特征图生成第二权重矩阵时,对上述第二局部特征图沿行维度(即水平维度)进行一维卷积处理,即在宽度方向上对每一列像素进行卷积操作,例如,第二局部特征图表示为(B,C,H,W),对上述第二局部特征图采用轻量级的一维卷积模块沿W维度进行卷积处理,生成上述第二局部特征图的第二权重矩阵(B,C,1,W)。在得到上述第二局部特征图的第二权重矩阵后,根据上述第二权重矩阵对上述第二局部特征图沿行维度进行循环卷积,提取上述第二局部特征图的全局特征,得到第二全局特征图。
可选地,在对上述第二局部特征图进行一维卷积处理以获取第二权重矩阵时,可与第一分支一样,采用深度可分离卷积进行处理,以减少参数量和计算量。例如,一维卷积模块采用如图所示的轻量级网络结构,其中,在进行平均池化时沿行维度进行平均池化,得到(B,C,1,W)形状的第二权重矩阵,1×3深度可分离卷积替换为3×1深度可分离卷积。
本申请实施例中,由于从不同维度采用一维的权重矩阵对局部特征图进行循环卷积处理,得到不同维度的全局特征,并将得到的全局特征进行拼接得到包含宽度方向和高度方向的全局特征,减少了提取局部特征图的全局特征时的计算量,同时,根据第一局部特征图和第二局部特征图分别动态提取其权重矩阵,再根据其权重矩阵对相应的局部特征图进行全局特征提取,使得卷积神经网络模型对于不同尺度的待识别图像具有良好的适应性,提高卷积神经网络模型的图像识别效果。
在一些实施例中,上述第二卷积模块还包括位置嵌入模块,上述步骤在将上述局部特征图沿通道方向进行拆分之前,还包括:
通过上述位置嵌入模块根据上述局部特征图的权重矩阵对上述局部特征图进行特征提取,得到位置特征图,并将上述位置特征图与上述局部特征图相加,得到包含位置特征的局部特征图。
可选地,在将根据局部特征图的权重矩阵对该局部特征图进行全局特征提取之前,通过位置嵌入模块对该局部特征图进行卷积处理,动态生成该局部特征图的权重矩阵并根据该权重矩阵对该局部特征图进行局部特征提取,生成位置特征图,并将该位置特征图与该局部特征图相加,得到嵌入了位置特征的局部特征图。
可选地,上述位置嵌入模块为二维卷积模块,对输入的样本图像进行卷积处理生成二维的位置特征图,即,使位置特征图的大小与输入的局部特征图的分辨率大小一致,以便直接将上述位置特征图与局部特征图相加,得到包含位置特征的局部特征图。例如,位置嵌入模块采用两层的轻量级卷积网络结构,即“卷积+归一化处理+激活函数+卷积”的简单结构,由于需要生成二维的位置特征图,因此,卷积层可以采用3×3的深度可分离卷积来对局部特征图进行卷积处理生成位置特征图。
本申请实施例中,由于图像的位置特征可以加强对图像内容的描述区分能力,在根据局部特征图提取全局特征之前,根据局部特征图的权重矩阵提取该局部特征图的位置特征,并将该位置特征嵌入到局部特征图中,使该局部特征图包含位置特征,因此,后续基于包含位置特征的全局特征图进行图像识别时可以提高识别准确度。
在一些实施例中,第二卷积模块的网络结构如图4所示,可包括位置嵌入模块、第一分支和第二分支,通过位置嵌入模块将局部特征图的位置特征嵌入局部特征图后,将嵌入了位置特征的局部特征图拆分,得到第一局部特征图和第二局部特征图并分别输入到第一分支和第二分支中,第一分支和第二分支分别基于行维度和列维度,根据第一局部特征图和第二局部特征图的权重矩阵对其进行全局特征提取,得到第一全局特征图和第二全局特征图,再将第一全局特征图和第二全局特征图进行拼接,得到全局特征图。此时,第一分支根据第一权重矩阵对第一局部特征图沿列维度进行循环卷积处理,提取第一局部特征图的全局特征时,上述循环卷积操作可表示为:
wH=f(w,H)
xP=x+f(pe,H)
其中,yi表示输出的第一全局特征的第i个元素,H是指第一局部特征图的行数(即高度),是指第一权重矩阵中的第k个权重(卷积核),其中wH是指若第一权重矩阵的长度不是H,则对第一权重矩阵进行插值操作,使其长度为H,(())H是指对第一局部特征图进行H点的周期延拓,是指对输入x(第一局部特征图)取第k+i个元素,xP是指加上位置特征的第一局部特征图,p即pe,代表与第一局部特征图对应的位置特征。
对应于上述基于卷积神经网络模型的图像识别方法,图5示出了本申请实施例提供的一种卷积神经网络模型训练方法的流程示意图,详述如下:
获取构建的卷积神经网络模型,并将样本图像输入到上述卷积神经网络进行训练,直至上述卷积神经网络满足预设要求,得到卷积神经网络模型。
可选地,根据用户需求(如需进行人脸识别任务,则卷积神经网络需要设置分类头对人脸图像进行分类识别)构建卷积神经网络,即根据用户需求设置卷积神经网络的网络结构(如可以基于现有的VGGNet、AlexNet的网络结构进行构建),并将构建的卷积神经网络进行初始。如,为了缓解卷积神经网络层数较深时存在的梯度消失和梯度爆炸的问题,基于残差网络ResNet18的网络结构构建卷积神经网络,其中,残差网络ResNet18包括17个卷积层和与1个全连接层。
具体地,获取预先构建的卷积神经网络,将相应的样本图像作为上述卷积神经网络的输入进行训练,直至上述卷积神经网络满足预设的要求(如卷积神经网络的识别准确度到达预设阈值,如0.988),则停止训练卷积神经网络,得到训练完成的卷积神经网络模型。其中,在进行卷积神经网络的训练时,卷积神经网络根据输入图像的权重矩阵来对输入图像进行全局特征提取,上述输入图像即输入的样本图像。
可选地,上述卷积神经网络根据输入图像的权重矩阵来对输入图像进行全局特征提取之前,通过卷积生成输入图像的权重矩阵,从而根据该权重矩阵对该输入图像进行全局特征的提取。需要指出的是,由于上述卷积神经网络采用动态权重,并不是固定的权重矩阵,因此,在卷积神经网络的训练过程中,根据损失函数对卷积神经网络的参数进行更新时,对权重矩阵进行更新时,是对卷积生成输入图像的权重矩阵的卷积模块的参数进行更新。
可选地,上述样本图像为用户进行的图像识别任务所对应的、带标签的样本图像,以便直接采用上述样本图像进行训练,不需要再对样本图像进行标注。例如在训练用于进行人脸识别的卷积神经网络模型时,采用CelebA人脸属性数据集作为训练集,CelebA包含了10177个身份的202599张人脸图像,每张图像都作了特征标记,可用于作为卷积神经网络的输入对卷积神经网络进行训练。在采用样本图像进行训练时,可将部分样本图像作为训练集进行训练,将一部分样本图像作为验证集和测试集,以调整卷积神经网络,得到良好的卷积神经网络模型。
可选地,在将样本图像作为卷积神经网络的输入进行训练时,单次训练输入的样本图像可以是一张,也可以是多张,如100张,批量输入样本图像时,输入的样本图像的数量表示为批处理大小Batch Size,相应的,卷积神经网络提取到的样本图像的特征形状可以表示为四维格式(B,H,W,C),B表示批处理大小,H表示高度,W表示宽度,C表示通道channel。
本申请实施例中,根据用户需求构建卷积神经网络,将带标签的样本图像作为上述卷积神经网络的输入进行训练,直至上述卷积神经网络满足预设要求,得到训练完成的卷积神经网络模型,由于上述卷积神经网络包括用于根据输入图像的权重矩阵对输入图像进行全局特征提取的动态卷积模块,因此,在卷积神经网络训练过程中,根据各个样本图像所对应的权重矩阵对各样本图像进行特征提取,各个样本图像不需要采用同一个静态的权重矩阵,使得卷积神经网络在训练过程中能够很好地适应不同尺度的样本图像,减少对卷积神经网络进行多尺度训练的难度,并使得卷积神经网络模型对多尺度图像具备良好的识别效果。
对应于上述基于卷积神经网络模型的图像识别方法或卷积神经网络模型训练方法,以下基于部分应用场景对基于卷积神经网络模型的图像识别方法进行介绍。
(一)行人检测
行人检测一直是计算机视觉研究中的热点和难点,行人检测所需要解决的问题是:找出图像或视频帧中所有的行人,包括其位置和大小,一般用矩形框表示。行人检测可以与行人跟踪、行人重识别等技术结合,应用于无人驾驶、智能交通等领域,因此,行人检测技术具有很强的应用价值。由于行人检测所检测的待识别图像大小不一,而本申请提供的基于卷积神经网络模型的图像识别方法,正是针对图像尺寸大小不一的问题,通过动态获取各个待识别图像的权重矩阵对待识别图像进行全局特征的提取,从而进行行人检测,提高行人检测的准确度。
首先,将采集到的待识别图像输入卷积神经网络模型(行人检测模型),通过第一卷积模块对该待识别图像进行局部特征的提取,以获取待识别图像的边缘、角点、线等特征信息,得到局部特征图,再将该局部特征图作为第二卷积模块的输入。由于行人检测是检测图像中的行人,行人的特征(如身体、头)之间具有较强的空间关系,因此,通过位置嵌入模块对局部特征图进行卷积处理,得到该局部特征图的位置特征图,并将其与局部特征图相加,使局部特征图包含位置信息,再将该局部特征图沿通道方向进行拆分得到第一局部特征图和第二局部特征图,输入到第一分支和第二分支,沿列维度和行维度对第一局部特征图、第二局部特征图进行卷积处理,得到该待识别图像的权重矩阵,再根据权重矩阵对其相应的局部特征图进行卷积处理,得到第一局部特征图和第二局部特征图对应的第一全局特征图和第二全局特征图,并将其沿通道方向拼接,得到完整的全局特征图。识别模块获取第二卷积模块输出的全局特征图,根据该全局特征图检测存在的行人特征,从而输出相应的检测结果。
(二)目标检测
在目标检测任务中,待测图像和被测目标的大小经常是不固定的,例如自动驾驶相关的检测任务可能同时需要检测大卡车和动物,医疗病灶检测任务可能需要同时检测大小不一的病灶,在被测物体尺度相差较大时,通常通过获取不同尺寸的待识别图像检测不同大小的目标,模型通常难以很好地适应对应大小差距大的检测,此时,基于本申请提供的卷积神经网络模型,可以根据输入的待识别图像动态调整权重矩阵,根据各个待识别图像各自的权重矩阵对该待识别图像进行检测,实现较好的检测效果。
应理解,上述实施例中各步骤的序号的大小并不意味着执行顺序的先后,各过程的执行顺序应以其功能和内在逻辑确定,而不应对本申请实施例的实施过程构成任何限定。
实施例二:
对应于上文实施例所述的基于卷积神经网络模型的图像识别方法,图6示出了本申请实施例提供的图像识别装置的结构框图,为了便于说明,仅示出了与本申请实施例相关的部分。
参照图6,该装置包括:输入模块61、卷积神经网络模型62,上述卷积神经网络模型基于待识别图像的权重矩阵对上述待识别图像进行全局特征提取。其中,
输入模块61,用于将待识别图像输入到上述卷积神经网络模型;
卷积神经网络模型62,用于对上述待识别图像依次进行特征提取和识别,得到识别结果。
本申请实施例中,将待识别图像输入到经训练的卷积神经网络模型,通过上述卷积神经网络模型对上述待识别图像依次进行特征提取和识别,得到识别结果。由于上述卷积神经网络模型在特征提取的过程中,根据待识别图像的权重矩阵对该待识别图像进行全局特征的提取,卷积神经网络模型同时具有全局感受野和动态权重,因此,可以较好地对不同大小的待识别图像进行特征提取,从而提高卷积神经网络模型的识别准确度。
在一些实施例中,上述图像识别装置还包括:
待识别图像获取模块,用于获取待识别图像。
在一些实施例中,上述卷积神经网络模型62包括:
特征提取单元,用于对上述待识别图像进行特征提取;
识别单元,用于对提取到的特征进行识别,得到识别结果。
在一些实施例中,上述特征提取单元包括:
第一卷积单元,用于对上述待识别图像进行局部特征提取,得到局部特征图;
第二卷积单元,用于根据上述局部特征图的权重矩阵对上述局部特征图进行全局特征提取,得到全局特征图。
在一些实施例中,上述第二卷积单元包括:
拆分单元,用于将上述局部特征图沿通道方向拆分,得到第一局部特征图和第二局部特征图;
第一分支单元,用于根据第一局部特征图的权重矩阵对第一局部特征图进行全局特征提取,得到第一全局特征图;
第二分支单元,用于根据第二局部特征图的权重矩阵对第二局部特征图进行全局特征提取,得到第二全局特征图;
拼接单元,用于将上述第一全局特征图和上述第二全局特征图沿通道方向进行拼接,得到全局特征图。
在一些实施例中,上述第一分支单元包括:
第一权重单元,用于对第一局部特征图像进行卷积处理,生成第一权重矩阵;
全局特征提取单元,用于根据上述第一权重矩阵对第一局部特征图进行全局特征提取,得到第一全局特征图;
上述第二分支单元包括:
第二权重单元,用于对第二局部特征图像进行卷积处理,生成第二权重矩阵;
全局特征提取单元,用于根据上述第二权重矩阵对第二局部特征图进行全局特征提取,得到第二全局特征图。
在一些实施例中,上述第二卷积单元还包括:
位置嵌入单元,用于根据上述局部特征图的权重矩阵对上述局部特征图进行特征提取,得到位置特征图,并将上述位置特征图与上述局部特征图相加,得到包含位置特征的局部特征图。
对应于上文实施例所述的卷积神经网络模型训练方法,图7示出了本申请实施例提供的卷积神经网络模型训练装置的结构框图,参照图7,该装置包括:
训练模块71,用于获取构建的卷积神经网络模型,并将样本图像输入到上述卷积神经网络进行训练,直至上述卷积神经网络满足预设要求,得到卷积神经网络模型,其中,上述卷积神经网络基于样本图像的权重矩阵对所述样本图像进行全局特征提取。
本申请实施例中,根据用户需求构建卷积神经网络,将带标签的样本图像作为上述卷积神经网络的输入进行训练,直至上述卷积神经网络满足预设要求,得到训练完成的卷积神经网络模型,由于上述卷积神经网络包括用于根据输入图像的权重矩阵对输入图像进行全局特征提取的动态卷积模块,因此,在卷积神经网络训练过程中,根据各个样本图像所对应的权重矩阵对各样本图像进行特征提取,各个样本图像不需要采用同一个静态的权重矩阵,使得卷积神经网络在训练过程中能够很好地适应不同尺度的样本图像,减少对卷积神经网络进行多尺度训练的难度,并使得卷积神经网络模型对多尺度图像具备良好的识别效果。
需要说明的是,上述装置/单元之间的信息交互、执行过程等内容,由于与本申请方法实施例基于同一构思,其具体功能及带来的技术效果,具体可参见方法实施例部分,此处不再赘述。
实施例三:
图8为本申请一实施例提供的终端设备的结构示意图。如图8所示,该实施例的终端设备8包括:至少一个处理器80(图8中仅示出一个处理器)、存储器81以及存储在所述存储器81中并可在所述至少一个处理器80上运行的计算机程序82,所述处理器80执行所述计算机程序82时实现上述任意各个方法实施例中的步骤。
示例性的,所述计算机程序82可以被分割成一个或多个模块/单元,所述一个或者多个模块/单元被存储在所述存储器81中,并由所述处理器80执行,以完成本申请。所述一个或多个模块/单元可以是能够完成特定功能的一系列计算机程序指令段,该指令段用于描述所述计算机程序82在所述终端设备81中的执行过程。例如,所述计算机程序82可以被分割成输入模块61和卷积神经网络模型62,上述卷积神经网络模型基于待识别图像的权重矩阵对上述待识别图像进行全局特征提取,各模块之间具体功能如下:
输入模块61,用于将待识别图像输入到上述卷积神经网络模型;
卷积神经网络模型62,用于对上述待识别图像依次进行特征提取和识别,得到识别结果。
或者,所述计算机程序82可以被分割成训练模块71,模块具体功能如下:
训练模块71,用于获取构建的卷积神经网络模型,并将样本图像输入到上述卷积神经网络进行训练,直至上述卷积神经网络满足预设要求,得到卷积神经网络模型,其中,上述卷积神经网络基于样本图像的权重矩阵对所述样本图像进行全局特征提取。
所述终端设备8可以是桌上型计算机、笔记本、掌上电脑及云端服务器等计算设备。该终端设备可包括,但不仅限于,处理器80、存储器81。本领域技术人员可以理解,图8仅仅是终端设备8的举例,并不构成对终端设备8的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件,例如还可以包括输入输出设备、网络接入设备等。
所称处理器80可以是中央处理单元(Central Processing Unit,CPU),该处理器80还可以是其他通用处理器、数字信号处理器(Digital Signal Processor,DSP)、专用集成电路(Application Specific Integrated Circuit,ASIC)、现场可编程门阵列(Field-Programmable Gate Array,FPGA)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
所述存储器81在一些实施例中可以是所述终端设备8的内部存储单元,例如终端设备8的硬盘或内存。所述存储器81在另一些实施例中也可以是所述终端设备8的外部存储设备,例如所述终端设备8上配备的插接式硬盘,智能存储卡(Smart Media Card,SMC),安全数字(Secure Digital,SD)卡,闪存卡(Flash Card)等。进一步地,所述存储器81还可以既包括所述终端设备8的内部存储单元也包括外部存储设备。所述存储器81用于存储操作系统、应用程序、引导装载程序(BootLoader)、数据以及其他程序等,例如所述计算机程序的程序代码等。所述存储器81还可以用于暂时地存储已经输出或者将要输出的数据。
所属领域的技术人员可以清楚地了解到,为了描述的方便和简洁,仅以上述各功能单元、模块的划分进行举例说明,实际应用中,可以根据需要而将上述功能分配由不同的功能单元、模块完成,即将所述装置的内部结构划分成不同的功能单元或模块,以完成以上描述的全部或者部分功能。实施例中的各功能单元、模块可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中,上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。另外,各功能单元、模块的具体名称也只是为了便于相互区分,并不用于限制本申请的保护范围。上述系统中单元、模块的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
本申请实施例还提供了一种网络设备,该网络设备包括:至少一个处理器、存储器以及存储在所述存储器中并可在所述至少一个处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现上述任意各个方法实施例中的步骤。
本申请实施例还提供了一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现可实现上述各个方法实施例中的步骤。
本申请实施例提供了一种计算机程序产品,当计算机程序产品在终端设备上运行时,使得终端设备执行时实现可实现上述各个方法实施例中的步骤。
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请实现上述实施例方法中的全部或部分流程,可以通过计算机程序来指令相关的硬件来完成,所述的计算机程序可存储于一计算机可读存储介质中,该计算机程序在被处理器执行时,可实现上述各个方法实施例的步骤。其中,所述计算机程序包括计算机程序代码,所述计算机程序代码可以为源代码形式、对象代码形式、可执行文件或某些中间形式等。所述计算机可读介质至少可以包括:能够将计算机程序代码携带到拍照装置/终端设备的任何实体或装置、记录介质、计算机存储器、只读存储器(ROM,Read-Only Memory)、随机存取存储器(RAM,RandomAccess Memory)、电载波信号、电信信号以及软件分发介质。例如U盘、移动硬盘、磁碟或者光盘等。在某些司法管辖区,根据立法和专利实践,计算机可读介质不可以是电载波信号和电信信号。
在上述实施例中,对各个实施例的描述都各有侧重,某个实施例中没有详述或记载的部分,可以参见其它实施例的相关描述。
本领域普通技术人员可以意识到,结合本文中所公开的实施例描述的各示例的单元及算法步骤,能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本申请的范围。
在本申请所提供的实施例中,应该理解到,所揭露的装置/网络设备和方法,可以通过其它的方式实现。例如,以上所描述的装置/网络设备实施例仅仅是示意性的,例如,所述模块或单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通讯连接可以是通过一些接口,装置或单元的间接耦合或通讯连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
以上所述实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围,均应包含在本申请的保护范围之内。
Claims (10)
1.一种基于卷积神经网络模型的图像识别方法,其特征在于,所述卷积神经网络模型基于待识别图像的权重矩阵对所述待识别图像进行全局特征提取;
所述图像识别方法包括:
将待识别图像输入到经训练的所述卷积神经网络模型,通过所述卷积神经网络模型对所述待识别图像依次进行特征提取和识别,得到识别结果。
2.如权利要求1所述的图像识别方法,其特征在于,所述卷积神经网络模型包括特征提取模块和识别模块,所述通过所述卷积神经网络模型对所述待识别图像依次进行特征提取和识别,得到识别结果,包括:
通过所述特征提取模块对所述待识别图像进行特征提取;
基于所述识别模块对提取到的特征进行识别,得到识别结果。
3.如权利要求2所述的图像识别方法,其特征在于,所述特征提取模块包括第一卷积模块和第二卷积模块,所述第二卷积模块基于待识别图像的权重矩阵对所述待识别图像进行全局特征提取,所述通过所述特征提取模块对所述待识别图像进行特征提取,包括:
基于所述第一卷积模块对所述待识别图像进行局部特征提取,得到局部特征图;
基于所述第二卷积模块根据所述局部特征图的权重矩阵对所述局部特征图进行全局特征提取,得到全局特征图。
4.如权利要求3所述的图像识别方法,其特征在于,所述第二卷积模块包括第一分支和第二分支,所述基于所述第二卷积模块根据所述局部特征图的权重矩阵对所述局部特征图进行全局特征提取,得到全局特征图,包括:
将所述局部特征图沿通道方向拆分,得到第一局部特征图和第二局部特征图,并将所述第一局部特征图和所述第二局部特征图分别输入到所述第一分支和所述第二分支;
所述第一分支和所述第二分支分别根据输入的局部特征图的权重矩阵对相应的局部特征图进行特征提取,得到第一全局特征图和第二全局特征图;
将所述第一全局特征图和所述第二全局特征图沿通道方向进行拼接,得到全局特征图。
5.如权利要求4所述的图像识别方法,其特征在于,所述第一分支和所述第二分支分别根据输入的局部特征图的权重矩阵对相应的局部特征图进行特征提取,得到第一全局特征图和第二全局特征图,包括:
根据所述第一局部特征图生成列维度的权重矩阵,得到第一权重矩阵,并根据所述第一权重矩阵对所述第一局部特征图沿列维度进行卷积处理,得到第一全局特征图;
根据所述第二分支的输入图像生成行维度的权重矩阵,得到第二权重矩阵,并根据所述第二权重矩阵对所述第二局部特征图沿行维度进行卷积处理,得到第二全局特征图。
6.如权利要求4所述的图像识别方法,其特征在于,所述第二卷积模块还包括位置嵌入模块,在所述将所述局部特征图沿通道方向进行拆分之前,还包括:
通过所述位置嵌入模块根据所述局部特征图的权重矩阵对所述局部特征图进行特征提取,得到位置特征图,并将所述位置特征图与所述局部特征图相加,得到包含位置特征的局部特征图。
7.一种卷积神经网络模型训练方法,其特征在于,包括:
获取构建的卷积神经网络模型,并将样本图像输入到所述卷积神经网络进行训练,直至所述卷积神经网络满足预设要求,得到卷积神经网络模型;
其中,所述卷积神经网络基于样本图像的权重矩阵对所述样本图像进行全局特征提取。
8.一种图像识别装置,其特征在于,包括:
输入模块和经训练的卷积神经网络模型,所述卷积神经网络模型基于待识别图像的权重矩阵对所述待识别图像进行全局特征提取;
所述输入模块,用于将待识别图像输入到所述卷积神经网络模型;
所述卷积神经网络模型,用于对所述待识别图像依次进行特征提取和识别,得到识别结果。
9.一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1至6任一项所述的图像识别方法或如权利要求7所述的卷积神经网络模型训练方法。
10.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至6任一项所述的图像识别方法或如权利要求7所述的卷积神经网络模型训练方法。
Priority Applications (2)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202211254852.4A CN115578590A (zh) | 2022-10-13 | 2022-10-13 | 基于卷积神经网络模型的图像识别方法、装置及终端设备 |
PCT/CN2022/142241 WO2024077781A1 (zh) | 2022-10-13 | 2022-12-27 | 基于卷积神经网络模型的图像识别方法、装置及终端设备 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202211254852.4A CN115578590A (zh) | 2022-10-13 | 2022-10-13 | 基于卷积神经网络模型的图像识别方法、装置及终端设备 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN115578590A true CN115578590A (zh) | 2023-01-06 |
Family
ID=84585951
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202211254852.4A Pending CN115578590A (zh) | 2022-10-13 | 2022-10-13 | 基于卷积神经网络模型的图像识别方法、装置及终端设备 |
Country Status (2)
Country | Link |
---|---|
CN (1) | CN115578590A (zh) |
WO (1) | WO2024077781A1 (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116309582A (zh) * | 2023-05-19 | 2023-06-23 | 之江实验室 | 一种便携式超声扫描图像的识别方法、装置及电子设备 |
CN116645365A (zh) * | 2023-07-21 | 2023-08-25 | 锋睿领创(珠海)科技有限公司 | 基于频谱的石英玻璃检测方法、装置、设备及介质 |
Family Cites Families (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108229379A (zh) * | 2017-12-29 | 2018-06-29 | 广东欧珀移动通信有限公司 | 图像识别方法、装置、计算机设备和存储介质 |
CN109726725B (zh) * | 2018-12-28 | 2021-05-04 | 中南大学 | 一种基于大间隔类间互异性多核学习的油画作者识别方法 |
CN112052886B (zh) * | 2020-08-21 | 2022-06-03 | 暨南大学 | 基于卷积神经网络的人体动作姿态智能估计方法及装置 |
CN112215840B (zh) * | 2020-10-30 | 2024-07-16 | 上海商汤临港智能科技有限公司 | 图像检测、行驶控制方法、装置、电子设备及存储介质 |
CN112101318A (zh) * | 2020-11-17 | 2020-12-18 | 深圳市优必选科技股份有限公司 | 基于神经网络模型的图像处理方法、装置、设备及介质 |
-
2022
- 2022-10-13 CN CN202211254852.4A patent/CN115578590A/zh active Pending
- 2022-12-27 WO PCT/CN2022/142241 patent/WO2024077781A1/zh unknown
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116309582A (zh) * | 2023-05-19 | 2023-06-23 | 之江实验室 | 一种便携式超声扫描图像的识别方法、装置及电子设备 |
CN116309582B (zh) * | 2023-05-19 | 2023-08-11 | 之江实验室 | 一种便携式超声扫描图像的识别方法、装置及电子设备 |
CN116645365A (zh) * | 2023-07-21 | 2023-08-25 | 锋睿领创(珠海)科技有限公司 | 基于频谱的石英玻璃检测方法、装置、设备及介质 |
CN116645365B (zh) * | 2023-07-21 | 2023-11-17 | 锋睿领创(珠海)科技有限公司 | 基于频谱的石英玻璃检测方法、装置、设备及介质 |
Also Published As
Publication number | Publication date |
---|---|
WO2024077781A1 (zh) | 2024-04-18 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112528878B (zh) | 检测车道线的方法、装置、终端设备及可读存储介质 | |
CN112446398B (zh) | 图像分类方法以及装置 | |
Izidio et al. | An embedded automatic license plate recognition system using deep learning | |
CN115240121B (zh) | 一种用于增强行人局部特征的联合建模方法和装置 | |
CN107527007B (zh) | 在车辆图像处理系统中检测关注对象的方法 | |
CN111767882A (zh) | 一种基于改进yolo模型的多模态行人检测方法 | |
CN109960742B (zh) | 局部信息的搜索方法及装置 | |
KR102140805B1 (ko) | 위성 영상의 물체 식별을 위한 뉴럴 네트워크 학습 방법 및 장치 | |
CN114359851A (zh) | 无人驾驶目标检测方法、装置、设备及介质 | |
CN115187844A (zh) | 基于神经网络模型的图像识别方法、装置及终端设备 | |
CN110176024B (zh) | 在视频中对目标进行检测的方法、装置、设备和存储介质 | |
CN113869282B (zh) | 人脸识别方法、超分模型训练方法及相关设备 | |
CN111368943A (zh) | 图像中对象的识别方法和装置、存储介质及电子装置 | |
CN110852311A (zh) | 一种三维人手关键点定位方法及装置 | |
CN114444565B (zh) | 一种图像篡改检测方法、终端设备及存储介质 | |
CN115578590A (zh) | 基于卷积神经网络模型的图像识别方法、装置及终端设备 | |
Ling et al. | Optimization of autonomous driving image detection based on RFAConv and triplet attention | |
CN112766012B (zh) | 二维码图像的识别方法、装置、电子设备及存储介质 | |
CN115272691A (zh) | 一种钢筋绑扎状态检测模型的训练方法、识别方法及设备 | |
CN111179270A (zh) | 基于注意力机制的图像共分割方法和装置 | |
CN115690488A (zh) | 基于卷积神经网络模型的图像识别方法、装置及终端设备 | |
CN115018039A (zh) | 一种神经网络蒸馏方法、目标检测方法以及装置 | |
CN111353429A (zh) | 基于眼球转向的感兴趣度方法与系统 | |
CN114758145A (zh) | 一种图像脱敏方法、装置、电子设备及存储介质 | |
CN110659631A (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 |