CN113705285A - 主体识别方法、装置、及计算机可读存储介质 - Google Patents

主体识别方法、装置、及计算机可读存储介质 Download PDF

Info

Publication number
CN113705285A
CN113705285A CN202010440237.7A CN202010440237A CN113705285A CN 113705285 A CN113705285 A CN 113705285A CN 202010440237 A CN202010440237 A CN 202010440237A CN 113705285 A CN113705285 A CN 113705285A
Authority
CN
China
Prior art keywords
picture
weight
subject
center point
determining
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
Application number
CN202010440237.7A
Other languages
English (en)
Inventor
陆瑾
熊龙飞
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Zhuhai Kingsoft Office Software Co Ltd
Wuhan Kingsoft Office Software Co Ltd
Original Assignee
Zhuhai Kingsoft Office Software Co Ltd
Wuhan Kingsoft Office Software Co Ltd
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by Zhuhai Kingsoft Office Software Co Ltd, Wuhan Kingsoft Office Software Co Ltd filed Critical Zhuhai Kingsoft Office Software Co Ltd
Priority to CN202010440237.7A priority Critical patent/CN113705285A/zh
Publication of CN113705285A publication Critical patent/CN113705285A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting

Abstract

一种主体识别方法、装置及计算机可读介质,获取待识别的图片;将所获取的待识别的图片输入到预先训练好的目标检测模型,检测出所述图片中的物体,及所述图片中每个物体的属性特征;所述每个物体的属性特征包括该物体的中心点位置、该物体的尺寸参数、及该物体的置信度;根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体。本申请能够灵活的识别图片中的目标主体。

Description

主体识别方法、装置、及计算机可读存储介质
技术领域
本文涉及计算机技术,尤指一种主体识别方法、装置、及计算机可读存储介质。
背景技术
现有技术中,图片中的主体大多是通过人为观察图像特征进行阈值分割,将前景和背景大致区分开。例如,通过二值化,得到黑白的二值图像,通过膨胀处理去除噪点,保留原图中主体部分;或者将图像转换色彩空间RGB转换成HSV,剥离H向量后,得到灰度向量,再使用直方图或者二值化算法将背景与前景区分开来,得到主体部分。这类方法需要对每张图片进行人为的设定阈值,无法自动识别到主体的类型,且在背景复杂或者存在多个前景的时候,无法有效分离出主体。
发明内容
本申请提供了一种主体识别方法、装置、及计算机可读存储介质,能够达到灵活的识别图片中的目标主体的目的。
本申请提供了一种主体识别方法,获取待识别的图片;将所获取的待识别的图片输入到预先训练好的目标检测模型,检测出所述图片中的物体,及所述图片中每个物体的属性特征;所述每个物体的属性特征包括该物体的中心点位置、该物体的尺寸参数、及该物体的置信度;根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体。
与相关技术相比,本申请实施例基于深度学习目标检测的主体识别方法,通过使用目标检测的技术替代人为设定阈值、二值化、膨胀等方法去分割前景背景的方法,能够灵活的识别图片中的目标主体,并适用于复杂的背景和存在多个前景的情况。
在一个示例性实施例中,本申请实施例通过设置属性特征各自对应的权重,能够指定不同的类别的物体作为目标主体。
本申请的其它特征和优点将在随后的说明书中阐述,并且,部分地从说明书中变得显而易见,或者通过实施本申请而了解。本申请的其他优点可通过在说明书以及附图中所描述的方案来实现和获得。
附图说明
附图用来提供对本申请技术方案的理解,并且构成说明书的一部分,与本申请的实施例一起用于解释本申请的技术方案,并不构成对本申请技术方案的限制。
图1为本申请实施例主体识别方法流程图;
图2为本申请实施例在图片中识别出的目标主体示意图;
图3为本申请实施例主体识别装置模块示意图。
具体实施方式
本申请描述了多个实施例,但是该描述是示例性的,而不是限制性的,并且对于本领域的普通技术人员来说显而易见的是,在本申请所描述的实施例包含的范围内可以有更多的实施例和实现方案。尽管在附图中示出了许多可能的特征组合,并在具体实施方式中进行了讨论,但是所公开的特征的许多其它组合方式也是可能的。除非特意加以限制的情况以外,任何实施例的任何特征或元件可以与任何其它实施例中的任何其他特征或元件结合使用,或可以替代任何其它实施例中的任何其他特征或元件。
本申请包括并设想了与本领域普通技术人员已知的特征和元件的组合。本申请已经公开的实施例、特征和元件也可以与任何常规特征或元件组合,以形成由权利要求限定的独特的发明方案。任何实施例的任何特征或元件也可以与来自其它发明方案的特征或元件组合,以形成另一个由权利要求限定的独特的发明方案。因此,应当理解,在本申请中示出和/或讨论的任何特征可以单独地或以任何适当的组合来实现。因此,除了根据所附权利要求及其等同替换所做的限制以外,实施例不受其它限制。此外,可以在所附权利要求的保护范围内进行各种修改和改变。
此外,在描述具有代表性的实施例时,说明书可能已经将方法和/或过程呈现为特定的步骤序列。然而,在该方法或过程不依赖于本文所述步骤的特定顺序的程度上,该方法或过程不应限于所述的特定顺序的步骤。如本领域普通技术人员将理解的,其它的步骤顺序也是可能的。因此,说明书中阐述的步骤的特定顺序不应被解释为对权利要求的限制。此外,针对该方法和/或过程的权利要求不应限于按照所写顺序执行它们的步骤,本领域技术人员可以容易地理解,这些顺序可以变化,并且仍然保持在本申请实施例的精神和范围内。
如图1所示,本申请实施例主体识别方法,包括如下操作:
S1、获取待识别的图片;
S2、将所获取的待识别的图片输入到预先训练好的目标检测模型,检测出所述图片中的物体,及所述图片中每个物体的属性特征;
其中,所述每个物体的属性特征包括该物体的中心点位置、该物体的尺寸参数、及该物体的置信度等等。在其他实施方式中,物体的属性特征也可以不限于前述种类,可以根据具体情形进行增加。
其中,上述置信度是指目标检测模型预测的该物体属于哪个类型的概率。比如目标物体属于人的概率、动物的概率、物品的概率等等。
S3、根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体。
本申请实施例能够较灵活的识别图片中的目标主体。
在一个示例性实施例中,上述预训练好的目标检测模型,采用基于深度学习的目标检测算法实现,能够检测到一张图片中包含的物体、人的位置信息等等。可以使用FasterRCNN、Mask RCNN、SSD、或者YOLOv3等检测模型,采用图片对应各类类别(人物、桌子、电脑等)(coco数据集)以及物体的在图片中的矩形坐标框出来。
在一个示例性实施例中,上述操作S3中所述根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体,包括:根据所检测出的所述图片中每个物体的属性特征,以及物体的每个属性特征各自对应的预设权重,确定所述图片中作为目标主体的物体。
本申请实施例通过设置权重,能够指定不同的类别的物体作为目标主体对象进行区域识别,即,设定作为目标主体的要求后提取出目标主体,预先设定的主体物体的要求可以包括:一、单个主体占图片的面积;二、单个主体在图片中的位置;三、置信度,单个主体属于某个类别的概率。
在一个示例性实施例中,上述操作S3中所述根据所检测出的所述图片中每个物体的属性特征,以及物体的每个属性特征各自的预设权重,确定所述图片中作为目标主体的物体,包括如下操作:
S31、根据每个物体的尺寸参数分别确定该物体在所述图片中的面积占比,根据每个物体的中心点位置分别确定该物体的中心点与所述图片的中心点的距离;其中,图片的属性参数可以在本步骤之前获取。
S32、根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定所述图片中作为目标主体的物体。
在一个示例性实施例中,预先训练好的目标检测神经网络检测出图片中各类型的单个物体N个,并且提取图片的属性特征和每个物体的属性特征。其中,每个物体的属性特征包括尺寸参数属性特征,表示为[xi,yi,wi,hi,pi],0<=i<N;xi、yi表示每个物体中心点;wi、hi表示每个物体的宽和高;置信度属性特征表示为pi。其中,图片的属性特征包括尺寸参数属性特征,表示为[X,Y,W,H];X、Y表示图片的中心点,W、H表示图片的宽和高。
在一个示例性实施例中,每个物体所述图片中的面积占比A,可以采用如下公式获取:
Figure BDA0002503808740000051
在一个示例性实施例中,每个物体的中心点与所述图片的中心点的距离B,可以采用如下公式获取:
Figure BDA0002503808740000052
上述计算每个物体的面积占比和中心点与所述图片的中心点的距离不限于上述计算形式,可以采用别的计算式,在此不做限定。
在一个示例性实施例中,上述操作S32中所述根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定所述图片中作为目标主体的物体,包括如下操作:
S321、根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定每个物体作为目标主体的物体的得分;
S322、将得分值大于预定阈值的物体确定为目标主体。
在一个示例性实施例中,上述操作S321中所述根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定每个物体作为目标主体的物体的得分,包括:
对于所述图片中的每个物体分别进行如下操作:
S3211、根据该物体的面积占比和预设的尺寸参数的权重、中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重对应相乘并求和;
S3212、根据求和结果得到该物体作为目标主体的得分值。
在一个示例性实施例中,每个物体作为目标主体的得分值,可以采用如下方式获取;
将N个检测到的目标物体依次进行计算:
Figure BDA0002503808740000061
其中α表示预设的尺寸参数的权重,本实施例中为面积占比权重。β表示预设的中心点位置的权重。φ表示置信度权重。各个权重的大小可以根据实际需求进行设置,例如,需要检测占比面积最大的目标物体作为目标主体,则α值权重加大。最后,选取检测到的单个物体N个物体中Obj最大的物体i作为目标主体目标,完成主体检测识别。
在一个示例性实施例中,上述操作S3中所述确定所述图片中作为目标主体的物体后还包括:将所确定的作为目标主体的物体从所述图片中单独截取。
如图2所示,以一个实际图片为例对上述方法进行详细说明。将图2中的图片输入预训练好的目标检测模型,识别出:
图片属性:
[302,212,604,424]。
单个目标的属性,
狗:[131,307,144,86,0.96];
人:[224 222 84 275 0.92];
马:[523 234 194 214 0.85]。
计算每个物体作为目标主体的得分值:
Figure BDA0002503808740000062
Figure BDA0002503808740000071
其中Obj1表示狗作为目标主体的得分;Obj2表示人作为目标主体的得分;Obj3表示马作为目标主体的得分。
当α=0,β=0.1φ=0.9时:
Figure BDA0002503808740000072
Figure BDA0002503808740000073
Figure BDA0002503808740000074
所以obj1最大,因此主体选择为置信度优先的狗目标,作为目标主体。
当α=0,β=0.5φ=0.1时:
Figure BDA0002503808740000075
Figure BDA0002503808740000076
Figure BDA0002503808740000077
通过obj的值,选择最靠近中心点的人obj2作为目标主体。
当α=0.8,β=0.1φ=0.1时:
Figure BDA0002503808740000078
Figure BDA0002503808740000079
Figure BDA00025038087400000710
通过obj的值,选取obj3马为主体,因为权重设置侧重与主体所占的面积。
本申请实施例通过设置属性特征各自对应的权重,能够指定不同的类别的物体作为目标主体。
在一个示例实施例中,上述操作S3中所述根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体,可以根据每个物体的尺寸参数分别确定该物体在所述图片中的面积占比,根据每个物体的中心点位置分别确定该物体的中心点与所述图片的中心点的距离;根据每个物体的面积占比、每个物体的中心点与图片的中心点的距离、以及置信度,确定所述图片中作为目标主体的物体,计算方式不同之处仅将上述的权重去除,在此不再赘述。
本申请通过使用目标检测的技术替代人为设定阈值、二值化、膨胀等方法去分割前景背景的方法,能够适用于复杂的背景和存在多个前景的情况,且能够指定不同的类别的物体作为目标主体对象进行区域识别。并且能够根据需要的主体,通过检测到的所有目标的区域面积大小、靠近中心位置以及是属于什么类别物体,来提取出主体。
如图3所示,本申请实施例主体识别装置,包括如下模块:
获取模块10,用于获取待识别的图片;检测模块20,用于将所获取的待识别的图片输入到预先训练好的目标检测模型,检测出所述图片中的物体,及所述图片中每个物体的属性特征;所述每个物体的属性特征包括该物体的中心点位置、该物体的尺寸参数、及该物体的置信度;确定模块30,用于根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体。
本申请实施例提供主体识别装置,包括处理器和存储器,其特征在于,所述存储器中存储有用于进行主体识别的程序;所述处理器用于读取所述用于主体识别的程序,执行上述中任一项所述的方法。
本申请实施例提供一种计算机可读介质,用于存储用于进行主体识别的程序,所述程序在被执行时执行上述中任一项所述的方法。
本领域普通技术人员可以理解,上文中所公开方法中的全部或某些步骤、系统、装置中的功能模块/单元可以被实施为软件、固件、硬件及其适当的组合。在硬件实施方式中,在以上描述中提及的功能模块/单元之间的划分不一定对应于物理组件的划分;例如,一个物理组件可以具有多个功能,或者一个功能或步骤可以由若干物理组件合作执行。某些组件或所有组件可以被实施为由处理器,如数字信号处理器或微处理器执行的软件,或者被实施为硬件,或者被实施为集成电路,如专用集成电路。这样的软件可以分布在计算机可读介质上,计算机可读介质可以包括计算机存储介质(或非暂时性介质)和通信介质(或暂时性介质)。如本领域普通技术人员公知的,术语计算机存储介质包括在用于存储信息(诸如计算机可读指令、数据结构、程序模块或其他数据)的任何方法或技术中实施的易失性和非易失性、可移除和不可移除介质。计算机存储介质包括但不限于RAM、ROM、EEPROM、闪存或其他存储器技术、CD-ROM、数字多功能盘(DVD)或其他光盘存储、磁盒、磁带、磁盘存储或其他磁存储装置、或者可以用于存储期望的信息并且可以被计算机访问的任何其他的介质。此外,本领域普通技术人员公知的是,通信介质通常包含计算机可读指令、数据结构、程序模块或者诸如载波或其他传输机制之类的调制数据信号中的其他数据,并且可包括任何信息递送介质。

Claims (10)

1.一种主体识别方法,包括:
获取待识别的图片;
将所获取的待识别的图片输入到预先训练好的目标检测模型,检测出所述图片中的物体,及所述图片中每个物体的属性特征;所述每个物体的属性特征包括该物体的中心点位置、该物体的尺寸参数、及该物体的置信度;
根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体。
2.根据权利要求1所述的主体识别方法,其特征在于,所述根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体,包括:
根据所检测出的所述图片中每个物体的属性特征,以及物体的每个属性特征各自对应的预设权重,确定所述图片中作为目标主体的物体。
3.根据权利要求2所述的主体识别方法,其特征在于,所述根据所检测出的所述图片中每个物体的属性特征,以及物体的每个属性特征各自对应的预设权重,确定所述图片中作为目标主体的物体包括:
根据每个物体的尺寸参数分别确定该物体在所述图片中的面积占比,根据每个物体的中心点位置分别确定该物体的中心点与所述图片的中心点的距离;
根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定所述图片中作为目标主体的物体。
4.根据权利要求3所述的主体识别方法,其特征在于,所述根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定所述图片中作为目标主体的物体,包括:
根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定每个物体作为目标主体的物体的得分;
将得分值大于预定阈值的物体确定为目标主体。
5.根据权利要求4所述的主体识别方法,其特征在于,所述根据每个物体的面积占比和预设的尺寸参数的权重、每个物体的中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重,确定每个物体作为目标主体的物体的得分,包括:
对于所述图片中的每个物体分别进行如下操作:
将该物体的面积占比和预设的尺寸参数的权重、中心点与图片的中心点的距离和预设的中心点位置的权重、以及置信度和置信度的权重对应相乘并求和;
根据求和结果得到该物体作为目标主体的得分值。
6.根据权利要求1所述的主体识别方法,其特征在于,所述确定所述图片中作为目标主体的物体后还包括:
将所确定的作为目标主体的物体从所述图片中单独截取。
7.根据权利要求1所述的主体识别方法,其特征在于,所述预先训练好的目标检测模型为Faster RCNN、Mask RCNN、SSD、或者YOLOv3等模型。
8.一种主体识别装置,其特征在于,包括:
获取模块,用于获取待识别的图片;
检测模块,用于将所获取的待识别的图片输入到预先训练好的目标检测模型,检测出所述图片中的物体,及所述图片中每个物体的属性特征;所述每个物体的属性特征包括该物体的中心点位置、该物体的尺寸参数、及该物体的置信度;
确定模块,用于根据所检测出的所述图片中每个物体的属性特征,确定所述图片中作为目标主体的物体。
9.一种主体识别装置,包括处理器和存储器,其特征在于,所述存储器中存储有用于进行主体识别的程序;所述处理器用于读取所述用于主体识别的程序,执行权利要求1-8中任一项所述的方法。
10.一种计算机可读介质,用于存储用于进行主体识别的程序,所述程序在被执行时执行权利要求1-8中任一项所述的方法。
CN202010440237.7A 2020-05-22 2020-05-22 主体识别方法、装置、及计算机可读存储介质 Pending CN113705285A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202010440237.7A CN113705285A (zh) 2020-05-22 2020-05-22 主体识别方法、装置、及计算机可读存储介质

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202010440237.7A CN113705285A (zh) 2020-05-22 2020-05-22 主体识别方法、装置、及计算机可读存储介质

Publications (1)

Publication Number Publication Date
CN113705285A true CN113705285A (zh) 2021-11-26

Family

ID=78646179

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202010440237.7A Pending CN113705285A (zh) 2020-05-22 2020-05-22 主体识别方法、装置、及计算机可读存储介质

Country Status (1)

Country Link
CN (1) CN113705285A (zh)

Citations (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN108960290A (zh) * 2018-06-08 2018-12-07 Oppo广东移动通信有限公司 图像处理方法、装置、计算机可读存储介质和电子设备
CN109871730A (zh) * 2017-12-05 2019-06-11 杭州海康威视数字技术股份有限公司 一种目标识别方法、装置及监控设备
CN110276767A (zh) * 2019-06-28 2019-09-24 Oppo广东移动通信有限公司 图像处理方法和装置、电子设备、计算机可读存储介质
CN110473185A (zh) * 2019-08-07 2019-11-19 Oppo广东移动通信有限公司 图像处理方法和装置、电子设备、计算机可读存储介质
WO2020019966A1 (zh) * 2018-07-27 2020-01-30 阿里巴巴集团控股有限公司 一种检测方法及装置、一种计算设备及存储介质

Patent Citations (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN109871730A (zh) * 2017-12-05 2019-06-11 杭州海康威视数字技术股份有限公司 一种目标识别方法、装置及监控设备
CN108960290A (zh) * 2018-06-08 2018-12-07 Oppo广东移动通信有限公司 图像处理方法、装置、计算机可读存储介质和电子设备
WO2020019966A1 (zh) * 2018-07-27 2020-01-30 阿里巴巴集团控股有限公司 一种检测方法及装置、一种计算设备及存储介质
CN110276767A (zh) * 2019-06-28 2019-09-24 Oppo广东移动通信有限公司 图像处理方法和装置、电子设备、计算机可读存储介质
CN110473185A (zh) * 2019-08-07 2019-11-19 Oppo广东移动通信有限公司 图像处理方法和装置、电子设备、计算机可读存储介质

Similar Documents

Publication Publication Date Title
CN108460356B (zh) 一种基于监控系统的人脸图像自动处理系统
KR102641115B1 (ko) 객체 검출을 위한 영상 처리 방법 및 장치
CN107346409B (zh) 行人再识别方法和装置
US8737740B2 (en) Information processing apparatus, information processing method, and non-transitory computer-readable storage medium
US9092868B2 (en) Apparatus for detecting object from image and method therefor
JP6192271B2 (ja) 画像処理装置、画像処理方法及びプログラム
JP4098021B2 (ja) シーン識別方法および装置ならびにプログラム
JP6756406B2 (ja) 画像処理装置、画像処理方法および画像処理プログラム
WO2019102608A1 (en) Image processing device, image processing method, and image processing program
CN112784712B (zh) 一种基于实时监控的失踪儿童预警实现方法、装置
CN112836625A (zh) 人脸活体检测方法、装置、电子设备
Zhu et al. Automatic object detection and segmentation from underwater images via saliency-based region merging
JP2017102622A (ja) 画像処理装置、画像処理方法及びプログラム
CN111814690A (zh) 一种目标重识别方法、装置和计算机可读存储介质
CN114255468A (zh) 一种笔迹识别方法及其相关设备
CN114299363A (zh) 图像处理模型的训练方法、图像分类方法及装置
CN113963295A (zh) 视频片段中地标识别方法、装置、设备及存储介质
JP2021071769A (ja) 物体追跡装置および物体追跡方法
CN114119970B (zh) 目标跟踪方法及装置
CN113705285A (zh) 主体识别方法、装置、及计算机可读存储介质
CN111414952B (zh) 行人重识别的噪声样本识别方法、装置、设备和存储介质
JPH11110542A (ja) パターン抽出方法および装置、そのプログラムを記録した媒体
CN114022509A (zh) 基于多个动物的监控视频的目标跟踪方法及相关设备
CN107992853B (zh) 人眼检测方法、装置、计算机设备及存储介质
CN110785769A (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