CN112434631A - 目标物识别方法、装置、电子设备及可读存储介质 - Google Patents
目标物识别方法、装置、电子设备及可读存储介质 Download PDFInfo
- Publication number
- CN112434631A CN112434631A CN202011386786.7A CN202011386786A CN112434631A CN 112434631 A CN112434631 A CN 112434631A CN 202011386786 A CN202011386786 A CN 202011386786A CN 112434631 A CN112434631 A CN 112434631A
- Authority
- CN
- China
- Prior art keywords
- image
- target object
- sample set
- sample
- target
- 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 50
- 238000012545 processing Methods 0.000 claims abstract description 58
- 238000003709 image segmentation Methods 0.000 claims abstract description 44
- 230000003313 weakening effect Effects 0.000 claims abstract description 31
- 238000010606 normalization Methods 0.000 claims abstract description 18
- 230000006870 function Effects 0.000 claims description 29
- 230000008569 process Effects 0.000 claims description 19
- 238000012549 training Methods 0.000 claims description 16
- 238000004364 calculation method Methods 0.000 claims description 10
- 238000005286 illumination Methods 0.000 claims description 9
- 230000011218 segmentation Effects 0.000 claims description 8
- 238000004458 analytical method Methods 0.000 claims description 4
- 239000000126 substance Substances 0.000 claims description 2
- 241001465754 Metazoa Species 0.000 description 15
- 238000010586 diagram Methods 0.000 description 10
- 238000005070 sampling Methods 0.000 description 6
- 108091026890 Coding region Proteins 0.000 description 4
- 238000011176 pooling Methods 0.000 description 4
- 238000004422 calculation algorithm Methods 0.000 description 3
- 230000002829 reductive effect Effects 0.000 description 3
- 241001310793 Podium Species 0.000 description 2
- 238000004891 communication Methods 0.000 description 2
- 238000004590 computer program Methods 0.000 description 2
- 238000005516 engineering process Methods 0.000 description 2
- 230000000670 limiting effect Effects 0.000 description 2
- 239000004973 liquid crystal related substance Substances 0.000 description 2
- 230000003287 optical effect Effects 0.000 description 2
- 238000013473 artificial intelligence Methods 0.000 description 1
- 230000005540 biological transmission Effects 0.000 description 1
- 230000008094 contradictory effect Effects 0.000 description 1
- 238000013500 data storage Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 230000010365 information processing Effects 0.000 description 1
- 230000003993 interaction Effects 0.000 description 1
- 230000007246 mechanism Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000003068 static effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/20—Scenes; Scene-specific elements in augmented reality scenes
-
- 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
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/20—Image preprocessing
- G06V10/26—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion
- G06V10/267—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion by performing operations on regions, e.g. growing, shrinking or watersheds
-
- 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/20—Image preprocessing
- G06V10/32—Normalisation of the pattern dimensions
-
- 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/07—Target detection
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Multimedia (AREA)
- Data Mining & Analysis (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Biology (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- General Engineering & Computer Science (AREA)
- Bioinformatics & Computational Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Image Analysis (AREA)
Abstract
本发明涉及智能决策,揭露一种目标物识别方法,包括:对待识别图像执行尺度归一化处理,得到第一图像;将第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对第二图像执行背景弱化处理;将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果。本发明还提供一种目标物识别装置、电子设备及可读存储介质。本发明提高了目标物识别准确度。
Description
技术领域
本发明涉及智能决策领域,尤其涉及一种目标物识别方法、装置、电子设备及可读存储介质。
背景技术
随着科技的发展,人工智能在人们生活中的应用越来越广泛,例如,采用模型识别图像中动物的种类,当前通常直接将待识别图像输入至识别模型进行识别,然而,由于不同图像的背景复杂度不同、光照强度不同,导致识别结果不够准确。因此,亟需一种目标物识别方法,以提高目标物识别准确度。
发明内容
鉴于以上内容,有必要提供一种目标物识别方法,旨在提高目标物识别准确度。
本发明提供的目标物识别方法,包括:
解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像;
将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理;
将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
可选的,所述图像分割模型的训练过程包括:
从图像库中抽取第一数量的携带第一标注的第一样本,得到第一样本集;
对所述第一样本集执行数据增强处理,得到第二样本集;
将所述第二样本集输入所述图像分割模型进行训练,通过最小化第一损失函数和第二损失函数调整所述图像分割模型的模型参数,得到训练好的图像分割模型。
可选的,所述对所述第一样本集执行数据增强处理,得到第二样本集,包括:
将所述第一样本集中第二数量的图像样本随机旋转预设角度,得到第三图像集;
对所述第一样本集中第三数量的图像样本执行模糊处理,得到第四图像集;
对所述第一样本集中第四数量的图像样本执行光照增强处理,得到第五图像集;
将所述第一样本集中第五数量的图像样本随机裁剪为不同尺寸的图像,得到第六图像集;
将所述第一、第三、第四、第五及第六图像集中的图像样本汇总,并将汇总得到的图像样本中的每张图像样本修正为预设尺寸的图像,得到第二样本集。
可选的,所述第一损失函数的计算公式为:
其中,H为第一损失值,pij为第二样本集中第i个样本中的第j个像素点的真实值,qij为第二样本集中第i个样本中的第j个像素点的预测值,e为第二样本集中样本的总数量,f为第二样本集中第i个样本中像素点的总数量;
所述第二损失函数的计算公式为:
其中,S为第二损失值,α、β为常数,且α+β=1,(TP)i为第二样本集中第i个样本的预测目标物区域预测正确的像素点总数量,(FP)i为第二样本集中第i个样本的预测目标物区域预测错误的像素点总数量,(FN)i为第二样本集中第i个样本的真实目标物区域被预测错误的像素点总数量,e为第二样本集中样本的总数量。
可选的,所述对所述第二图像执行背景弱化处理,包括;
若所述第二图像中有一个目标物,则将所述第二图像中所述目标物的区域以外的区域用预设像素值替换,得到目标图像;
若所述第二图像中有多个目标物,则按照目标物的数量对所述第二图像进行复制,得到初始图像集,分别将所述多个目标物中的每个目标物的区域作为所述初始图像集中一张图像的目标区域,将每张图像中所述目标区域以外的区域用预设像素值替换,得到目标图像集。
可选的,所述图像识别模型的训练过程,包括:
从所述图像库中抽取第六数量的携带第二标注的第二样本,得到第七样本集;
将所述第七样本集输入至所述训练好的图像分割模型,得到第八样本集;
对所述第八样本集执行数据增强处理,得到第九样本集;
将所述第九样本集输入所述图像识别模型进行训练,通过最小化第三损失函数确定所述图像识别模型的模型参数,得到训练好的图像识别模型。
为了解决上述问题,本发明还提供一种目标物识别装置,所述装置包括:
解析模块,用于解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像;
分割模块,用于将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理;
识别模块,用于将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
可选的,所述对所述第二图像执行背景弱化处理,包括;
若所述第二图像中有一个目标物,则将所述第二图像中所述目标物的区域以外的区域用预设像素值替换,得到目标图像;
若所述第二图像中有多个目标物,则按照目标物的数量对所述第二图像进行复制,得到初始图像集,分别将所述多个目标物中的每个目标物的区域作为所述初始图像集中一张图像的目标区域,将每张图像中所述目标区域以外的区域用预设像素值替换,得到目标图像集。
为了解决上述问题,本发明还提供一种电子设备,所述电子设备包括:
至少一个处理器;以及,
与所述至少一个处理器通信连接的存储器;其中,
所述存储器存储有可被所述至少一个处理器执行的目标物识别程序,所述目标物识别程序被所述至少一个处理器执行,以使所述至少一个处理器能够执行上述目标物识别方法。
为了解决上述问题,本发明还提供一种计算机可读存储介质,所述计算机可读存储介质上存储有目标物识别程序,所述目标物识别程序可被一个或者多个处理器执行,以实现上述目标物识别方法。
相较现有技术,本发明首先对待识别图像执行尺度归一化处理,得到第一图像,本步骤通过执行尺度归一化处理得到目标尺寸的第一图像,可便于后续的图像分割及图像识别;然后,将第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对第二图像执行背景弱化处理,本步骤对第二图像执行背景弱化处理,降低了背景对图像中目标物识别的影响,使得后续的目标物识别结果更为准确;最后,将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果。因此,本发明提高了目标物识别准确度。
附图说明
图1为本发明一实施例提供的目标物识别方法的流程示意图;
图2为本发明一实施例提供的目标物识别装置的模块示意图;
图3为本发明一实施例提供的实现目标物识别方法的电子设备的结构示意图;
本发明目的的实现、功能特点及优点将结合实施例,参照附图做进一步说明。
具体实施方式
为了使本发明的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本发明进行进一步详细说明。应当理解,此处所描述的具体实施例仅用以解释本发明,并不用于限定本发明。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
需要说明的是,在本发明中涉及“第一”、“第二”等的描述仅用于描述目的,而不能理解为指示或暗示其相对重要性或者隐含指明所指示的技术特征的数量。由此,限定有“第一”、“第二”的特征可以明示或者隐含地包括至少一个该特征。另外,各个实施例之间的技术方案可以相互结合,但是必须是以本领域普通技术人员能够实现为基础,当技术方案的结合出现相互矛盾或无法实现时应当认为这种技术方案的结合不存在,也不在本发明要求的保护范围之内。
本发明提供一种目标物识别方法。参照图1所示,为本发明一实施例提供的目标物识别方法的流程示意图。该方法可以由一个电子设备执行,该电子设备可以由软件和/或硬件实现。
本实施例中,目标物识别方法包括:
S1、解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像。
本实施例中,所述目标物识别请求可以是人脸识别请求,也可以是动植物种类识别请求、桌子类别识别请求(餐桌、办公桌、讲台等桌子类别)或其它识别请求。
本实施例以目标物识别请求为动物种类识别请求为例进行说明,所述待识别图像中可以有一个动物,也可以有多个动物。
所述对所述待识别图像执行尺度归一化处理包括:
将待识别图像的尺寸归一化到预设尺寸(例如,513*513),归一化规则为:首先将待识别图像的长边缩放到513,根据长边与长边最终缩放的尺寸513求出缩放比例;然后根据缩放比例缩放短边,短边尺寸不足513时,对不足的部分外扩补0。
S2、将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理。
本实施例中,所述图像分割模型为deeplabv3+(语义分割模型),包括编码区及解码区。所述编码区包括第一残差模块及空洞卷积模块,所述第一残差模块为下采样16倍的resnet101网络,所述空洞卷积模块包括不同空洞率的卷积层;所述解码区包括上采样模块,所述上采样模块包括16倍的上采样层,解码区通过融合低层级的编码特征图和高层级的编码特征图,使得输出的特征图更为精细。
所述图像分割模型的训练过程包括步骤A11-A13:
A11、从图像库中抽取第一数量的携带第一标注的第一样本,得到第一样本集;
所述第一标注为人工或其它方式标注的第一样本中目标物(动物)的标签,第一样本中待识别目标物(动物)区域对应的标签为1,其它区域(背景)对应的标签为0,标签的尺寸为h*w*1,和原图一样大。
A12、对所述第一样本集执行数据增强处理,得到第二样本集;
所述对所述第一样本集执行数据增强处理,得到第二样本集,包括步骤B11-B15:
B11、将所述第一样本集中第二数量的图像样本随机旋转预设角度(例如,90度),得到第三图像集;
B12、对所述第一样本集中第三数量的图像样本执行模糊处理,得到第四图像集;
本实施例中,可采用高斯模糊算法对图像样本执行模糊处理。
B13、对所述第一样本集中第四数量的图像样本执行光照增强处理,得到第五图像集;
本实施例中,可通过给图像样本增加模拟光照来实现光照增强处理。
B14、将所述第一样本集中第五数量的图像样本随机裁剪为不同尺寸的图像,得到第六图像集;
B15、将所述第一、第三、第四、第五及第六图像集中的图像样本汇总,并将汇总得到的图像样本中的每张图像样本修正为预设尺寸的图像,得到第二样本集。
本实施例中,所述预设尺寸为513*513,对于尺寸不足513*513的图像样本,长边缩放到513,根据长边与长边最后缩放的尺寸513求出缩放比例,然后根据缩放比例缩放短边;对于短边尺寸不足513的图像样本,对不足的短边部分外扩以达到513,外扩部分像素值可设置为0即黑色。
A13、将所述第二样本集输入所述图像分割模型进行训练,通过最小化第一损失函数和第二损失函数调整所述图像分割模型的模型参数,得到训练好的图像分割模型。
所述第一损失函数的计算公式为:
其中,H为第一损失值,pij为第二样本集中第i个样本中的第j个像素点的真实值,qij为第二样本集中第i个样本中的第j个像素点的预测值,e为第二样本集中样本的总数量,f为第二样本集中第i个样本中像素点的总数量;
所述第二损失函数的计算公式为:
其中,S为第二损失值,α、β为常数,且α+β=1,(TP)i为第二样本集中第i个样本的预测目标物区域预测正确的像素点总数量,(FP)i为第二样本集中第i个样本的预测目标物区域预测错误的像素点总数量,(FN)i为第二样本集中第i个样本的真实目标物区域被预测错误的像素点总数量,e为第二样本集中样本的总数量。
该损失S是从预测全局的角度来期望FN、FP越小越好,同时通过使β大一点、α小一点(β=0.75,α=0.25)可以使得图像分割模型更偏向于FN越小越好,最终使得图像分割模型的召回率很高。
本实施例中,通过最小化第一、第二损失函数可达到保持精度的同时有较高的召回率;同时,通过调整α、β的值可为精度和召回率设置不同的重要级。
将第一图像输入至训练好的图像分割模型,可分割得到含有目标物和背景的第二图像,第二图像的shape为513*513*2(h*w*c),其中,第二图像中某个像素点是属于目标物(动物)区域还是背景区域,需要根据c通道的值来判断,如果c通道中背景的概率值大,则该像素点为背景区域,否则该素点为目标物区域。
本实施例中,所述对所述第二图像执行背景弱化处理,包括;
C11、若所述第二图像中有一个目标物,则将所述第二图像中所述目标物的区域以外的区域用预设像素值替换,得到目标图像;
C12、若所述第二图像中有多个目标物,则按照目标物的数量对所述第二图像进行复制,得到初始图像集,分别将所述多个目标物中的每个目标物的区域作为所述初始图像集中一张图像的目标区域,将每张图像中所述目标区域以外的区域用预设像素值替换,得到目标图像集。
例如,若图像分割模型输出的第二图像中有3个目标物,则复制后得到的初始图像集中共3张图像,可将第一个目标物的区域作为第一张图像的目标区域,将目标区域以外的区域设置为黑色,然后,将第二个目标物的区域作为第二张图像的目标区域,将目标区域以外的区域设置为黑色,以此类推,得到的目标图像集的三张图像中分别有一个目标物区域。
S3、将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
本实施例中,所述图像识别模型包括第二残差模块、全局池化模块及全连接模块,所述第二残差模块包括resnet50网络及注意力模块cbam,所述第二残差模块用于提取特征,所述全局池化模块包括GlobalAveragePooling网络,所述全连接模块包括两个全连接层和一个softmax网络。
所述图像识别模型的训练过程,包括:
D11、从所述图像库中抽取第六数量的携带第二标注的第二样本,得到第七样本集;
本实施例中,第二样本中的目标物数量为一个,第二标注为第二样本中的目标物的类别。
D12、将所述第七样本集输入至所述训练好的图像分割模型,得到第八样本集;
D13、对所述第八样本集执行数据增强处理,得到第九样本集;
D14、将所述第九样本集输入所述图像识别模型进行训练,通过最小化第三损失函数确定所述图像识别模型的模型参数,得到训练好的图像识别模型。
所述第三损失函数的计算公式为:
其中,L为第三损失值,M为第九样本集中目标物类别的总数量,yic为指示变量(0或1),表示第九样本集中第i个样本中目标物的预测类别与实际类别是否一致,一致时为1,不一致时为0,pic为第九样本集中第i个样本中目标物预测为第c类的概率,N为第九样本集中样本的总数量。
将背景弱化处理得到的图像(目标图像或目标图像集)输入所述训练好的图像识别模型可得到每张图像中目标物的种类。
由上述实施例可知,本发明提出的目标物识别方法,首先,对待识别图像执行尺度归一化处理,得到第一图像,本步骤通过执行尺度归一化处理得到目标尺寸的第一图像,可便于后续的图像分割及图像识别;然后,将第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对第二图像执行背景弱化处理,本步骤对第二图像执行背景弱化处理,降低了背景对图像中目标物识别的影响,使得后续的目标物识别结果更为准确;最后,将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果。因此,本发明提高了目标物识别准确度。
如图2所示,为本发明一实施例提供的目标物识别装置的模块示意图。
本发明所述目标物识别装置100可以安装于电子设备中。根据实现的功能,所述目标物识别装置100可以包括解析模块110、分割模块120及识别模块130。本发明所述模块也可以称之为单元,是指一种能够被电子设备处理器所执行,并且能够完成固定功能的一系列计算机程序段,其存储在电子设备的存储器中。
在本实施例中,关于各模块/单元的功能如下:
解析模块110,用于解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像。
本实施例中,所述目标物识别请求可以是人脸识别请求,也可以是动植物种类识别请求、桌子类别识别请求(餐桌、办公桌、讲台等桌子类别)或其它识别请求。
本实施例以目标物识别请求为动物种类识别请求为例进行说明,所述待识别图像中可以有一个动物,也可以有多个动物。
所述对所述待识别图像执行尺度归一化处理包括:
将待识别图像的尺寸归一化到预设尺寸(例如,513*513),归一化规则为:首先将待识别图像的长边缩放到513,根据长边与长边最终缩放的尺寸513求出缩放比例;然后根据缩放比例缩放短边,短边尺寸不足513时,对不足的部分外扩补0。
分割模块120,用于将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理。
本实施例中,所述图像分割模型为deeplabv3+(语义分割模型),包括编码区及解码区。所述编码区包括第一残差模块及空洞卷积模块,所述第一残差模块为下采样16倍的resnet101网络,所述空洞卷积模块包括不同空洞率的卷积层;所述解码区包括上采样模块,所述上采样模块包括16倍的上采样层,解码区通过融合低层级的编码特征图和高层级的编码特征图,使得输出的特征图更为精细。
所述图像分割模型的训练过程包括步骤A21-A23:
A21、从图像库中抽取第一数量的携带第一标注的第一样本,得到第一样本集;
所述第一标注为人工或其它方式标注的第一样本中目标物(动物)的标签,第一样本中待识别目标物(动物)区域对应的标签为1,其它区域(背景)对应的标签为0,标签的尺寸为h*w*1,和原图一样大。
A22、对所述第一样本集执行数据增强处理,得到第二样本集;
所述对所述第一样本集执行数据增强处理,得到第二样本集,包括步骤B21-B25:
B21、将所述第一样本集中第二数量的图像样本随机旋转预设角度(例如,90度),得到第三图像集;
B22、对所述第一样本集中第三数量的图像样本执行模糊处理,得到第四图像集;
本实施例中,可采用高斯模糊算法对图像样本执行模糊处理。
B23、对所述第一样本集中第四数量的图像样本执行光照增强处理,得到第五图像集;
本实施例中,可通过给图像样本增加模拟光照来实现光照增强处理。
B24、将所述第一样本集中第五数量的图像样本随机裁剪为不同尺寸的图像,得到第六图像集;
B25、将所述第一、第三、第四、第五及第六图像集中的图像样本汇总,并将汇总得到的图像样本中的每张图像样本修正为预设尺寸的图像,得到第二样本集。
本实施例中,所述预设尺寸为513*513,对于尺寸不足513*513的图像样本,长边缩放到513,根据长边与长边最后缩放的尺寸513求出缩放比例,然后根据缩放比例缩放短边;对于短边尺寸不足513的图像样本,对不足的短边部分外扩以达到513,外扩部分像素值可设置为0即黑色。
A23、将所述第二样本集输入所述图像分割模型进行训练,通过最小化第一损失函数和第二损失函数调整所述图像分割模型的模型参数,得到训练好的图像分割模型。
所述第一损失函数的计算公式为:
其中,H为第一损失值,pij为第二样本集中第i个样本中的第j个像素点的真实值,qij为第二样本集中第i个样本中的第j个像素点的预测值,e为第二样本集中样本的总数量,f为第二样本集中第i个样本中像素点的总数量;
所述第二损失函数的计算公式为:
其中,S为第二损失值,α、β为常数,且α+β=1,(TP)i为第二样本集中第i个样本的预测目标物区域预测正确的像素点总数量,(FP)i为第二样本集中第i个样本的预测目标物区域预测错误的像素点总数量,(FN)i为第二样本集中第i个样本的真实目标物区域被预测错误的像素点总数量,e为第二样本集中样本的总数量。
该损失S是从预测全局的角度来期望FN、FP越小越好,同时通过使β大一点、α小一点(β=0.75,α=0.25)可以使得图像分割模型更偏向于FN越小越好,最终使得图像分割模型的召回率很高。
本实施例中,通过最小化第一、第二损失函数可达到保持精度的同时有较高的召回率;同时,通过调整α、β的值可为精度和召回率设置不同的重要级。
将第一图像输入至训练好的图像分割模型,可分割得到含有目标物和背景的第二图像,第二图像的shape为513*513*2(h*w*c),其中,第二图像中某个像素点是属于目标物(动物)区域还是背景区域,需要根据c通道的值来判断,如果c通道中背景的概率值大,则该像素点为背景区域,否则该素点为目标物区域。
本实施例中,所述对所述第二图像执行背景弱化处理,包括;
C21、若所述第二图像中有一个目标物,则将所述第二图像中所述目标物的区域以外的区域用预设像素值替换,得到目标图像;
C22、若所述第二图像中有多个目标物,则按照目标物的数量对所述第二图像进行复制,得到初始图像集,分别将所述多个目标物中的每个目标物的区域作为所述初始图像集中一张图像的目标区域,将每张图像中所述目标区域以外的区域用预设像素值替换,得到目标图像集。
例如,若图像分割模型输出的第二图像中有3个目标物,则复制后得到的初始图像集中共3张图像,可将第一个目标物的区域作为第一张图像的目标区域,将目标区域以外的区域设置为黑色,然后,将第二个目标物的区域作为第二张图像的目标区域,将目标区域以外的区域设置为黑色,以此类推,得到的目标图像集的三张图像中分别有一个目标物区域。
识别模块130,用于将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
本实施例中,所述图像识别模型包括第二残差模块、全局池化模块及全连接模块,所述第二残差模块包括resnet50网络及注意力模块cbam,所述第二残差模块用于提取特征,所述全局池化模块包括GlobalAveragePooling网络,所述全连接模块包括两个全连接层和一个softmax网络。
所述图像识别模型的训练过程,包括:
D21、从所述图像库中抽取第六数量的携带第二标注的第二样本,得到第七样本集;
本实施例中,第二样本中的目标物数量为一个,第二标注为第二样本中的目标物的类别。
D22、将所述第七样本集输入至所述训练好的图像分割模型,得到第八样本集;
D23、对所述第八样本集执行数据增强处理,得到第九样本集;
D24、将所述第九样本集输入所述图像识别模型进行训练,通过最小化第三损失函数确定所述图像识别模型的模型参数,得到训练好的图像识别模型。
所述第三损失函数的计算公式为:
其中,L为第三损失值,M为第九样本集中目标物类别的总数量,yic为指示变量(0或1),表示第九样本集中第i个样本中目标物的预测类别与实际类别是否一致,一致时为1,不一致时为0,pic为第九样本集中第i个样本中目标物预测为第c类的概率,N为第九样本集中样本的总数量。
将背景弱化处理得到的图像(目标图像或目标图像集)输入所述训练好的图像识别模型可得到每张图像中目标物的种类。
由上述实施例可知,本发明提出的目标物识别装置100,首先,对待识别图像执行尺度归一化处理,得到第一图像,本步骤通过执行尺度归一化处理得到目标尺寸的第一图像,可便于后续的图像分割及图像识别;然后,将第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对第二图像执行背景弱化处理,本步骤对第二图像执行背景弱化处理,降低了背景对图像中目标物识别的影响,使得后续的目标物识别结果更为准确;最后,将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果。因此,本发明提高了目标物识别准确度。
如图3所示,为本发明一实施例提供的实现目标物识别方法的电子设备的结构示意图。
所述电子设备1是一种能够按照事先设定或者存储的指令,自动进行数值计算和/或信息处理的设备。所述电子设备1可以是计算机、也可以是单个网络服务器、多个网络服务器组成的服务器组或者基于云计算的由大量主机或者网络服务器构成的云,其中云计算是分布式计算的一种,由一群松散耦合的计算机集组成的一个超级虚拟计算机。
在本实施例中,电子设备1包括,但不仅限于,可通过系统总线相互通信连接的存储器11、处理器12、网络接口13,该存储器11中存储有目标物识别程序10,所述目标物识别程序10可被所述处理器12执行。图3仅示出了具有组件11-13以及目标物识别程序10的电子设备1,本领域技术人员可以理解的是,图3示出的结构并不构成对电子设备1的限定,可以包括比图示更少或者更多的部件,或者组合某些部件,或者不同的部件布置。
其中,存储器11包括内存及至少一种类型的可读存储介质。内存为电子设备1的运行提供缓存;可读存储介质可为如闪存、硬盘、多媒体卡、卡型存储器(例如,SD或DX存储器等)、随机访问存储器(RAM)、静态随机访问存储器(SRAM)、只读存储器(ROM)、电可擦除可编程只读存储器(EEPROM)、可编程只读存储器(PROM)、磁性存储器、磁盘、光盘等的非易失性存储介质。在一些实施例中,可读存储介质可以是电子设备1的内部存储单元,例如该电子设备1的硬盘;在另一些实施例中,该非易失性存储介质也可以是电子设备1的外部存储设备,例如电子设备1上配备的插接式硬盘,智能存储卡(Smart Media Card,SMC),安全数字(Secure Digital,SD)卡,闪存卡(Flash Card)等。本实施例中,存储器11的可读存储介质通常用于存储安装于电子设备1的操作系统和各类应用软件,例如存储本发明一实施例中的目标物识别程序10的代码等。此外,存储器11还可以用于暂时地存储已经输出或者将要输出的各类数据。
处理器12在一些实施例中可以是中央处理器(Central Processing Unit,CPU)、控制器、微控制器、微处理器、或其他数据处理芯片。该处理器12通常用于控制所述电子设备1的总体操作,例如执行与其他设备进行数据交互或者通信相关的控制和处理等。本实施例中,所述处理器12用于运行所述存储器11中存储的程序代码或者处理数据,例如运行目标物识别程序10等。
网络接口13可包括无线网络接口或有线网络接口,该网络接口13用于在所述电子设备1与客户端(图中未画出)之间建立通信连接。
可选的,所述电子设备1还可以包括用户接口,用户接口可以包括显示器(Display)、输入单元比如键盘(Keyboard),可选的用户接口还可以包括标准的有线接口、无线接口。可选的,在一些实施例中,显示器可以是LED显示器、液晶显示器、触控式液晶显示器以及OLED(Organic Light-Emitting Diode,有机发光二极管)触摸器等。其中,显示器也可以适当的称为显示屏或显示单元,用于显示在电子设备1中处理的信息以及用于显示可视化的用户界面。
应该了解,所述实施例仅为说明之用,在专利申请范围上并不受此结构的限制。
所述电子设备1中的所述存储器11存储的目标物识别程序10是多个指令的组合,在所述处理器12中运行时,可以实现:
解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像;
将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理;
将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
具体地,所述处理器12对上述目标物识别程序10的具体实现方法可参考图1对应实施例中相关步骤的描述,在此不赘述。需要强调的是,为进一步保证上述待识别图像的私密和安全性,上述待识别图像还可以存储于一区块链的节点中。
进一步地,所述电子设备1集成的模块/单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。所述计算机可读介质可以是非易失性的,也可以是非易失性的。所述计算机可读介质可以包括:能够携带所述计算机程序代码的任何实体或装置、记录介质、U盘、移动硬盘、磁碟、光盘、计算机存储器、只读存储器(ROM,Read-Only Memory)。
所述计算机可读存储介质上存储有目标物识别程序10,所述目标物识别程序10可被一个或者多个处理器执行,本发明计算机可读存储介质具体实施方式与上述目标物识别方法各实施例基本相同,在此不作赘述。
在本发明所提供的几个实施例中,应该理解到,所揭露的设备,装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述模块的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式。
所述作为分离部件说明的模块可以是或者也可以不是物理上分开的,作为模块显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。
另外,在本发明各个实施例中的各功能模块可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用硬件加软件功能模块的形式实现。
对于本领域技术人员而言,显然本发明不限于上述示范性实施例的细节,而且在不背离本发明的精神或基本特征的情况下,能够以其他的具体形式实现本发明。
因此,无论从哪一点来看,均应将实施例看作是示范性的,而且是非限制性的,本发明的范围由所附权利要求而不是上述说明限定,因此旨在将落在权利要求的等同要件的含义和范围内的所有变化涵括在本发明内。不应将权利要求中的任何附关联图标记视为限制所涉及的权利要求。
本发明所指区块链是分布式数据存储、点对点传输、共识机制、加密算法等计算机技术的新型应用模式。区块链(Blockchain),本质上是一个去中心化的数据库,是一串使用密码学方法相关联产生的数据块,每一个数据块中包含了一批次网络交易的信息,用于验证其信息的有效性(防伪)和生成下一个区块。区块链可以包括区块链底层平台、平台产品服务层以及应用服务层等。
此外,显然“包括”一词不排除其他单元或步骤,单数不排除复数。系统权利要求中陈述的多个单元或装置也可以由一个单元或装置通过软件或者硬件来实现。第二等词语用来表示名称,而并不表示任何特定的顺序。
最后应说明的是,以上实施例仅用以说明本发明的技术方案而非限制,尽管参照较佳实施例对本发明进行了详细说明,本领域的普通技术人员应当理解,可以对本发明的技术方案进行修改或等同替换,而不脱离本发明技术方案的精神和范围。
Claims (10)
1.一种目标物识别方法,其特征在于,所述方法包括:
解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像;
将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理;
将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
2.如权利要求1所述的目标物识别方法,其特征在于,所述图像分割模型的训练过程包括:
从图像库中抽取第一数量的携带第一标注的第一样本,得到第一样本集;
对所述第一样本集执行数据增强处理,得到第二样本集;
将所述第二样本集输入所述图像分割模型进行训练,通过最小化第一损失函数和第二损失函数调整所述图像分割模型的模型参数,得到训练好的图像分割模型。
3.如权利要求2所述的目标物识别方法,其特征在于,所述对所述第一样本集执行数据增强处理,得到第二样本集,包括:
将所述第一样本集中第二数量的图像样本随机旋转预设角度,得到第三图像集;
对所述第一样本集中第三数量的图像样本执行模糊处理,得到第四图像集;
对所述第一样本集中第四数量的图像样本执行光照增强处理,得到第五图像集;
将所述第一样本集中第五数量的图像样本随机裁剪为不同尺寸的图像,得到第六图像集;
将所述第一、第三、第四、第五及第六图像集中的图像样本汇总,并将汇总得到的图像样本中的每张图像样本修正为预设尺寸的图像,得到第二样本集。
4.如权利要求2所述的目标物识别方法,其特征在于,所述第一损失函数的计算公式为:
其中,H为第一损失值,pij为第二样本集中第i个样本中的第j个像素点的真实值,qij为第二样本集中第i个样本中的第j个像素点的预测值,e为第二样本集中样本的总数量,f为第二样本集中第i个样本中像素点的总数量;
所述第二损失函数的计算公式为:
其中,S为第二损失值,α、β为常数,且α+β=1,(TP)i为第二样本集中第i个样本的预测目标物区域预测正确的像素点总数量,(FP)i为第二样本集中第i个样本的预测目标物区域预测错误的像素点总数量,(FN)i为第二样本集中第i个样本的真实目标物区域被预测错误的像素点总数量,e为第二样本集中样本的总数量。
5.如权利要求1-4任一项所述的目标物识别方法,其特征在于,所述对所述第二图像执行背景弱化处理,包括;
若所述第二图像中有一个目标物,则将所述第二图像中所述目标物的区域以外的区域用预设像素值替换,得到目标图像;
若所述第二图像中有多个目标物,则按照目标物的数量对所述第二图像进行复制,得到初始图像集,分别将所述多个目标物中的每个目标物的区域作为所述初始图像集中一张图像的目标区域,将每张图像中所述目标区域以外的区域用预设像素值替换,得到目标图像集。
6.如权利要求1所述的目标物识别方法,其特征在于,所述图像识别模型的训练过程,包括:
从所述图像库中抽取第六数量的携带第二标注的第二样本,得到第七样本集;
将所述第七样本集输入至所述训练好的图像分割模型,得到第八样本集;
对所述第八样本集执行数据增强处理,得到第九样本集;
将所述第九样本集输入所述图像识别模型进行训练,通过最小化第三损失函数确定所述图像识别模型的模型参数,得到训练好的图像识别模型。
7.一种目标物识别装置,其特征在于,所述装置包括:
解析模块,用于解析用户基于客户端发出的目标物识别请求,获取所述请求携带的待识别图像,对所述待识别图像执行尺度归一化处理,得到第一图像;
分割模块,用于将所述第一图像输入至训练好的图像分割模型,分割得到含有目标物和背景的第二图像,对所述第二图像执行背景弱化处理;
识别模块,用于将背景弱化处理得到的图像输入至训练好的图像识别模型,得到目标物识别结果,将所述目标物识别结果反馈给所述客户端。
8.如权利要求7所述的目标物识别装置,其特征在于,所述对所述第二图像执行背景弱化处理,包括;
若所述第二图像中有一个目标物,则将所述第二图像中所述目标物的区域以外的区域用预设像素值替换,得到目标图像;
若所述第二图像中有多个目标物,则按照目标物的数量对所述第二图像进行复制,得到初始图像集,分别将所述多个目标物中的每个目标物的区域作为所述初始图像集中一张图像的目标区域,将每张图像中所述目标区域以外的区域用预设像素值替换,得到目标图像集。
9.一种电子设备,其特征在于,所述电子设备包括:
至少一个处理器;以及,
与所述至少一个处理器通信连接的存储器;其中,
所述存储器存储有可被所述至少一个处理器执行的目标物识别程序,所述目标物识别程序被所述至少一个处理器执行,以使所述至少一个处理器能够执行如权利要求1至6中任一项所述的目标物识别方法。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有目标物识别程序,所述目标物识别程序可被一个或者多个处理器执行,以实现如权利要求1至6任一项所述的目标物识别方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011386786.7A CN112434631A (zh) | 2020-12-01 | 2020-12-01 | 目标物识别方法、装置、电子设备及可读存储介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011386786.7A CN112434631A (zh) | 2020-12-01 | 2020-12-01 | 目标物识别方法、装置、电子设备及可读存储介质 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN112434631A true CN112434631A (zh) | 2021-03-02 |
Family
ID=74699093
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202011386786.7A Pending CN112434631A (zh) | 2020-12-01 | 2020-12-01 | 目标物识别方法、装置、电子设备及可读存储介质 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112434631A (zh) |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110490073A (zh) * | 2019-07-15 | 2019-11-22 | 浙江省北大信息技术高等研究院 | 目标检测方法、装置、设备及存储介质 |
CN110991454A (zh) * | 2019-12-23 | 2020-04-10 | 云南大学 | 一种叶片图像识别方法、装置、电子设备及存储介质 |
CN111340137A (zh) * | 2020-03-26 | 2020-06-26 | 上海眼控科技股份有限公司 | 图像识别方法、装置及存储介质 |
WO2020143309A1 (zh) * | 2019-01-09 | 2020-07-16 | 平安科技(深圳)有限公司 | 分割模型训练方法、oct图像分割方法、装置、设备及介质 |
US20200372268A1 (en) * | 2019-05-23 | 2020-11-26 | Baidu Online Network Technology (Beijing) Co., Ltd. | Method and apparatus for recognizing boundary of traffic sign |
-
2020
- 2020-12-01 CN CN202011386786.7A patent/CN112434631A/zh active Pending
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2020143309A1 (zh) * | 2019-01-09 | 2020-07-16 | 平安科技(深圳)有限公司 | 分割模型训练方法、oct图像分割方法、装置、设备及介质 |
US20200372268A1 (en) * | 2019-05-23 | 2020-11-26 | Baidu Online Network Technology (Beijing) Co., Ltd. | Method and apparatus for recognizing boundary of traffic sign |
CN110490073A (zh) * | 2019-07-15 | 2019-11-22 | 浙江省北大信息技术高等研究院 | 目标检测方法、装置、设备及存储介质 |
CN110991454A (zh) * | 2019-12-23 | 2020-04-10 | 云南大学 | 一种叶片图像识别方法、装置、电子设备及存储介质 |
CN111340137A (zh) * | 2020-03-26 | 2020-06-26 | 上海眼控科技股份有限公司 | 图像识别方法、装置及存储介质 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111932482B (zh) | 图像中目标物的检测方法、装置、电子设备及存储介质 | |
CN111595850B (zh) | 切片缺陷检测方法、电子装置及可读存储介质 | |
CN114462412B (zh) | 实体识别方法、装置、电子设备及存储介质 | |
CN112597135A (zh) | 用户分类方法、装置、电子设备及可读存储介质 | |
CN111860377A (zh) | 基于人工智能的直播方法、装置、电子设备及存储介质 | |
CN113065609B (zh) | 图像分类方法、装置、电子设备及可读存储介质 | |
CN111402579A (zh) | 道路拥堵程度预测方法、电子装置及可读存储介质 | |
CN113342977B (zh) | 发票图像分类方法、装置、设备及存储介质 | |
CN112966687B (zh) | 图像分割模型训练方法、装置及通信设备 | |
CN113344433A (zh) | 产品匹配方法、装置、电子设备及可读存储介质 | |
CN116052848B (zh) | 一种医学成像质控的数据编码方法及系统 | |
CN113706252B (zh) | 产品推荐方法、装置、电子设备及存储介质 | |
CN113705686B (zh) | 图像分类方法、装置、电子设备及可读存储介质 | |
CN112434631A (zh) | 目标物识别方法、装置、电子设备及可读存储介质 | |
CN114706985A (zh) | 文本分类方法、装置、电子设备及存储介质 | |
CN116049644A (zh) | 特征筛选和聚类分箱方法、装置、电子设备及存储介质 | |
CN114281991A (zh) | 文本分类方法、装置、电子设备及存储介质 | |
CN113688239A (zh) | 少样本下的文本分类方法、装置、电子设备及存储介质 | |
CN112233194B (zh) | 医学图片优化方法、装置、设备及计算机可读存储介质 | |
CN112580505B (zh) | 网点开关门状态识别方法、装置、电子设备及存储介质 | |
CN113312482A (zh) | 问题分类方法、装置、电子设备及可读存储介质 | |
CN111695441B (zh) | 图像文档处理方法、装置及计算机可读存储介质 | |
CN113064984A (zh) | 意图识别方法、装置、电子设备及可读存储介质 | |
CN114581177A (zh) | 产品推荐方法、装置、设备及存储介质 | |
CN113807256A (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 |