一种目标检测结果的融合方法、装置、设备及介质与流程

文档序号:26589944发布日期:2021-09-10 20:34阅读:130来源:国知局
一种目标检测结果的融合方法、装置、设备及介质与流程

1.本公开涉及计算机技术领域,尤其涉及智能交通和深度学习技术领域,具体涉及一种目标检测结果的融合方法、装置、设备、及介质。


背景技术:

2.在智慧交通里,目标检测是必不可少的一个环节,检测结果的质量直接影响后续的操作。在目标检测的最后一步,相关技术通常会采用非极大值抑制(non

maximum suppression,nms)算法,从众多的候选框里筛选出尽可能准确的目标框。
3.其中,nms算法的具体实现是:首先按照各候选框的得分值对各候选框进行排序,根据排序结果选中得分最高的当前候选框;然后遍历其余的候选框,如果存在某一候选框与当前候选框的重叠面积大于设定的阈值,则删除此候选框;在完成对其余全部候选框的遍历后,重新从当前存在的未处理的候选框中选中得分最高的候选框,并重复上述过程,以得到最终的目标框。
4.然而,在某些极端场景中,如密集重叠场景或者低可见度场景等,通过nms算法筛选出来的目标框可能会造成目标对象被漏检。


技术实现要素:

5.本公开提供了一种目标检测结果的融合方法、装置、设备、及介质。
6.根据本公开的一方面,提供了一种目标检测结果的融合方法,包括:
7.训练得到至少两个目标检测模型;其中,不同目标检测模型的数据关注度不同;
8.根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值;
9.将待识别图像分别输入至各目标检测模型中,获取各目标检测模型输出的目标检测结果;
10.根据所述识别权重以及所述重叠度阈值,对各目标检测结果进行融合处理,得到与待识别图像对应的融合检测结果。
11.根据本公开的另一方面,提供了一种目标检测结果的融合装置,包括:
12.模型训练模块,用于训练得到至少两个目标检测模型;其中,不同目标检测模型的数据关注度不同;
13.权重获取模块,用于根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值;
14.检测结果获取模块,用于将待识别图像分别输入至各目标检测模型中,获取各目标检测模型输出的目标检测结果;
15.结果融合模块,用于根据所述识别权重以及所述重叠度阈值,对各目标检测结果进行融合处理,得到与待识别图像对应的融合检测结果。
16.根据本公开的另一方面,提供了一种电子设备,包括:
17.至少一个处理器;以及
18.与所述至少一个处理器通信连接的存储器;其中,
19.所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行本公开中任一实施例所述的方法。
20.根据本公开的另一方面,提供了一种存储有计算机指令的非瞬时计算机可读存储介质,所述计算机指令用于使所述计算机执行本公开中任一实施例所述的方法。
21.根据本公开的另一方面,提供了一种计算机程序产品,包括计算机程序,所述计算机程序在被处理器执行时实现本公开中任一实施例所述的方法。
22.本公开实施例的技术方案可以避免目标对象被漏检,提高目标检测结果的准确性。
23.应当理解,本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征,也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
24.图1是根据本公开实施例的一种目标检测结果的融合方法的流程示意图;
25.图2是根据本公开实施例的又一种目标检测结果的融合方法的流程示意图;
26.图3是根据本公开实施例的又一种目标检测结果的融合方法的流程示意图;
27.图4是根据本公开实施例的一种目标检测结果的融合装置的结构示意图;
28.图5是根据本公开实施例的目标检测结果的融合方法的电子设备的框图。
具体实施方式
29.以下结合附图对本公开的示范性实施例做出说明,其中包括本公开实施例的各种细节以助于理解,应当将它们认为仅仅是示范性的。因此,本领域普通技术人员应当认识到,可以对这里描述的实施例做出各种改变和修改,而不会背离本公开的范围和精神。同样,为了清楚和简明,以下的描述中省略了对公知功能和结构的描述。
30.图1是根据本公开实施例的一种目标检测结果的融合方法的流程示意图,本实施例适用于对多个目标检测模型的检测结果进行融合的情形,该方法可以通过目标检测结果的融合装置来执行,该装置可以通过软件和/或硬件的方式实现,并一般可以集成在具有数据处理功能的终端或者服务器中。具体的,参考图1,该方法具体包括如下步骤:
31.步骤110、训练得到至少两个目标检测模型;其中,不同目标检测模型的数据关注度不同。
32.在本实施例中,可以使用不同数据关注度的训练样本图像集对机器学习模型进行迭代训练,得到不同的目标检测模型,每个训练样本图像集中可以包括多种类型的样本图像。针对一个训练样本图像集而言,如果该训练样本图像集中包括某一类型的样本图像数量较多,则可以说明该训练样本图像集对该类型下的图像关注度较高,如果使用该训练样本图像集对机器学习模型训练后,得到的目标检测模型对该类型下的图像关注度也较高。由于不同的目标检测模型由不同数据关注度的训练样本图像集训练而成,因此,不同的目标检测模型的数据关注度也不同。
33.在一个具体的实施例中,假设有两个用于对机器学习模型进行训练的训练样本图
像集,其中,第一个训练样本图像集中包括100张行人样本图像,20张汽车样本图像,第二个训练样本图像集中包括20张行人样本图像,100张汽车样本图像,如果使用上述两个训练样本图像集分别对机器学习模型进行训练,则可以分别得到两个目标检测模型,其中第一个目标检测模型对行人图像的关注度较高,第二个目标检测模型对汽车图像的关注度较高。
34.步骤120、根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值。
35.在本实施例中,可选的,可以根据目标检测模型的数据关注度,或者目标检测模型对应的训练样本图像集,确定目标检测模型针对不同分类类型的识别权重。所述识别权重可以用于表征目标检测模型针对不同类型下的图像的识别性能的优劣势,其中,目标检测模型针对具体类型下的图像的识别权重越高,则可以说明该目标检测模型针对该类型下的图像的识别性能越好。
36.在一个具体的实施例中,假设某一目标检测模型对行人图像的关注度较高,对汽车图像的关注度较低,则可以将该目标检测模型针对行人图像的识别权重设置为较高的数值,将针对汽车图像的识别权重设置为较低的数值。
37.在另一个具体的实施例中,还可以根据目标检测模型对应的训练样本图像集中不同类型对应的样本图像数量,确定该目标检测模型针对不同分类类型的识别权重。例如,某目标检测模型对应的训练样本图像集中包括20张行人样本图像,80张汽车样本图像,则可以将该目标检测模型针对行人图像的识别权重设置为0.2,将针对汽车图像的识别权重设置为0.8。
38.在本实施例中,在使用不同数据关注度的训练样本图像集对机器学习模型进行迭代训练的过程中,可选的,机器学习模型会根据训练样本图像集中各样本图像对应的训练对象的尺寸,学习到用于区分不同目标检测结果的重叠度阈值,所述目标检测结果为目标检测模型针对待识别图像得到的检测结果。其中,如果两个目标检测结果之间的重叠度大于或者等于重叠度阈值,则可以认为这两个目标检测结果对应待识别图像中的同一个目标对象,反之,则认为这两个目标检测结果对应待识别图像中的不同的目标对象。
39.在一个具体的实施例中,假设有两个用于对机器学习模型进行训练的训练样本图像集,第一个训练样本图像集中包括100张行人样本图像,20张卡车样本图像,第二个训练样本图像集中包括20张行人样本图像,100张卡车样本图像,其中,行人样本图像对应的训练对象为行人,卡车样本图像对应的训练对象为卡车。
40.在使用第一个训练样本图像集对机器学习模型进行迭代训练的过程中,机器学习模型会学习到用于区分卡车与行人的重叠度阈值,由于行人的尺寸通常小于卡车的尺寸,并且第一个训练样本图像集中行人样本图像的数量大于卡车样本图像的数量,因此,通过第一个训练样本图像集训练得到的目标检测模型对应的重叠度阈值较小;反之,通过第二个训练样本图像集训练得到的目标检测模型对应的重叠度阈值较大。
41.在本实施例中,训练得到各目标检测模型后,可以根据各目标检测模型的数据关注度,以及对应的重叠度阈值,确定与各分类类型分别对应的重叠度阈值。例如,假设目标检测模型对行人图像的关注度较高,则可以将该目标检测模型对应的重叠度阈值,作为行人图像对应的重叠度阈值。
42.步骤130、将待识别图像分别输入至各目标检测模型中,获取各目标检测模型输出
的目标检测结果。
43.在本实施例中,将待识别图像分别输入至各目标检测模型之后,目标检测模型会检测到待识别图像中包括的多个目标对象,得到与各目标对象对应的目标框、目标分类类型以及识别置信度,多个目标对象对应的目标框、目标分类类型以及识别置信度可以作为目标检测模型输出的目标检测结果。所述识别置信度可以为目标检测模型检测到的目标对象属于目标分类类型的可信度。
44.步骤140、根据所述识别权重以及所述重叠度阈值,对各目标检测结果进行融合处理,得到与待识别图像对应的融合检测结果。
45.在本实施例中,获取到各目标检测模型输出的目标检测结果后,可以在各目标检测结果中获取目标分类类型相同的多个目标框,然后根据该目标分类类型对应的重叠度阈值,在上述多个目标框中筛选满足融合条件的目标框,并根据各目标检测模型针对目标分类类型的识别权重,以及各目标框对应的识别置信度,对满足融合条件的目标框进行融合处理,得到与待识别图像对应的融合检测结果。
46.在一个具体的实施例中,假设第一个目标检测模型输出的目标检测结果中包括行人图像对应的目标框a,第二个目标检测模型输出的目标检测结果中包括行人图像对应的目标框b,则可以判断目标框a与目标框b之间的重叠度是否大于或者等于行人图像对应的重叠度阈值,若是,则可以确定目标框a与目标框b满足融合条件,并将目标框a与目标框b进行融合处理,得到融合检测结果。在一个可选的实施方式中,如果第一个目标检测模型对行人图像的识别权重较高,第二个目标检测模型对行人图像的识别权重较低,则可以将目标框a的识别置信度作为融合检测结果的识别置信度。
47.本实施例中通过将目标检测结果中重叠度较高的目标框进行融合,可以避免目标对象被漏检;其次,通过使用不同数据关注度的训练样本图像集对机器学习模型进行迭代训练,可以得到细粒度的目标检测模型,通过结合各目标检测模型针对不同类型下的图像的识别性能的优劣势,以及与各分类类型分别对应的重叠度阈值,将各目标检测结果进行融合,可以提高融合检测结果的准确性,对于一些极端场景,如密集人群或者低可见度场景等,可以得到质量较高的检测结果,有效支持后续指挥交通的建设。
48.本公开实施例的技术方案,通过训练得到至少两个目标检测模型,根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值,然后将待识别图像分别输入至各目标检测模型中,并获取各目标检测模型输出的目标检测结果,最后根据识别权重以及重叠度阈值,对各目标检测结果进行融合处理,得到与待识别图像对应的融合检测结果的技术手段,可以避免目标对象被漏检,提高目标检测结果的准确性。
49.图2是根据本公开实施例的又一种目标检测结果的融合方法的流程示意图,本实施例是对上述技术方案的进一步细化,本实施例中的技术方案可以与上述一个或者多个实施方式结合。具体的,参考图2,该方法具体包括如下步骤:
50.步骤210、训练得到至少两个目标检测模型;其中,不同目标检测模型的数据关注度不同。
51.在本实施例的一个实施方式中,训练得到至少两个目标检测模型,包括:使用不同数据关注度的训练样本图像集,训练同一模型结构的机器学习模型,得到各所述目标检测
模型;或者使用不同数据关注度的训练样本图像集,分别训练不同模型结构的机器学习模型,得到各所述目标检测模型,其中,数据关注度与设定分类类型关联。
52.其中,可以使用不同数据关注度的训练样本图像集,对同一模型结构的机器学习模型进行训练,得到不同数据关注度的目标检测模型,这样设置的好处在于,可以减少目标检测模型的训练时间,提高目标检测模型的训练效率;或者,可以使用不同数据关注度的训练样本图像集,分别训练不同模型结构(例如模型框架以及网络结构等)的机器学习模型,得到各目标检测模型,这样设置的好处在于,可以保证各目标检测模型具有较强的差异性,通过使用不同的目标检测模型共同对待识别图像进行检测,可以提高检测结果的准确性。
53.其中,目标检测模型的数据关注度与设定分类类型关联,例如,目标检测模型对行人图像的关注度较高,则该目标检测模型的数据关注度与行人图像关联。
54.步骤220、将测试样本图像集输入至各目标检测模型中,获取各所述目标检测模型针对各测试样本图像输出的比对检测结果。
55.在本实施例中,测试样本图像中包括:至少一个标注框,以及与标注框对应的分类类型。其中,每个标注框对应测试样本图像中的一个目标对象,所述标注框对应的分类类型为,该标注框对应的目标对象的分类类型。
56.在将测试样本图像集输入至各目标检测模型后,目标检测模型会检测到测试样本图像中包括的多个目标对象,得到与各目标对象对应的目标框以及目标分类类型,多个目标对象对应的目标框以及目标分类类型可以作为目标检测模型输出的比对检测结果。
57.步骤230、将各所述比对检测结果分别与测试样本图像集进行比对,获取各目标检测模型针对不同分类类型的识别准确度。
58.在本实施例中,可以将目标检测模型针对各测试样本图像输出的目标框以及目标分类类型,与该测试样本图像中包括的标注框以及分类类型进行比对,根据比对结果确定目标检测模型针对不同分类类型的识别准确度。
59.在一个具体的实施例中,假设测试样本图像中包括10个行人图像对应的标注框,10个汽车图像对应的标注框,将该测试样本图像输入至目标检测模型后,目标检测模型输出的比对检测结果中包括8个行人图像对应的目标框,2个汽车图像对应的目标框,则可以认为该目标检测模型针对行人图像的识别准确度为80%,针对汽车图像的识别准确度为20%。
60.步骤240、根据各目标检测模型针对各分类类型的识别准确度,确定各目标检测模型针对不同分类类型的识别权重。
61.在本实施例中,可以将各目标检测模型针对各分类类型的识别准确度,直接确定为各目标检测模型针对不同分类类型的识别权重;或者还可以将各目标检测模型针对各分类类型的识别准确度,按照预设数据转换方法进行处理(例如归一化处理),得到各目标检测模型针对不同分类类型的识别权重。
62.这样设置的好处在于,通过使用测试样本图像集对目标检测模型进行测试,可以保证目标检测模型针对不同分类类型的识别权重结果更加准确,由此可以提高后续目标检测结果的准确性。
63.步骤250、在训练得到的各目标检测模型使用的模型参数中,筛选得到非极大值抑制nms算法所使用的模型重叠度阈值。
64.在本实施例中,在使用不同数据关注度的训练样本图像集对机器学习模型进行迭代训练的过程中,机器学习模型会根据各样本图像对应的训练对象的尺寸以及nms算法,学习到用于区分不同目标检测结果的重叠度阈值。
65.在训练得到目标检测模型后,可以在目标检测模型使用的模型参数中,筛选得到目标检测模型学习到的重叠度阈值(也即模型重叠度阈值)。
66.步骤260、根据各目标检测模型的模型重叠度阈值,以及目标检测模型的数据关注度与分类类型之间的关联关系,确定与各分类类型分别对应的重叠度阈值。
67.在本实施例的一个实施方式中,获取到各目标检测模型的模型重叠度阈值后,可以根据目标检测模型的数据关注度与分类类型之间的关联关系,直接将模型重叠度阈值作为关联的分类类型对应的重叠度阈值。
68.例如,目标检测模型对行人图像的关注度较高,则该目标检测模型的数据关注度与行人图像关联,获取到该目标检测模型的模型重叠度阈值后,可以直接将该模型重叠度阈值作为行人图像对应的重叠度阈值。
69.在本实施例的另一个实施方式中,获取到各目标检测模型的模型重叠度阈值后,可以根据目标检测模型的数据关注度与分类类型之间的关联关系,对模型重叠度阈值按照预设数据转换方法进行处理(例如归一化处理),得到关联的分类类型对应的重叠度阈值。
70.这样设置的好处在于,通过在训练得到的各目标检测模型使用的模型参数中,筛选得到nms算法所使用的模型重叠度阈值,可以保证模型重叠度阈值的训练结果更加准确;其次,根据各目标检测模型的模型重叠度阈值,以及目标检测模型的数据关注度与分类类型之间的关联关系,确定与各分类类型分别对应的重叠度阈值,可以保证不同分类类型对应不同的重叠度阈值,相比相关技术中多种分类类型对应同一个重叠度阈值而言,可以提高融合检测结果的准确性。
71.步骤270、将待识别图像分别输入至各目标检测模型中,获取各目标检测模型输出的目标检测结果。
72.步骤280、根据所述识别权重以及所述重叠度阈值,对各目标检测结果进行融合处理,得到与待识别图像对应的融合检测结果。
73.本公开实施例的技术方案,通过训练得到至少两个目标检测模型,将测试样本图像集输入至各目标检测模型中,获取各目标检测模型输出的比对检测结果,并将各比对检测结果分别与测试样本图像集进行比对,获取各目标检测模型针对不同分类类型的识别准确度,然后根据各目标检测模型针对各分类类型的识别准确度,确定各目标检测模型针对不同分类类型的识别权重,在训练得到的各目标检测模型使用的模型参数中,筛选得到nms算法所使用的模型重叠度阈值,并根据各目标检测模型的模型重叠度阈值,确定与各分类类型分别对应的重叠度阈值,最后将待识别图像分别输入至各目标检测模型中,获取目标检测结果,根据识别权重以及重叠度阈值,对各目标检测结果进行融合处理,得到融合检测结果的技术手段,可以提供一种通用的、标准化的检索服务,可以避免目标对象被漏检,提高目标检测结果的准确性。
74.本公开实施例是对上述实施例的进一步细化,与上述实施例相同或相应的术语解释,本公开实施例不再赘述。图3为本公开实施例提供的一种目标检测结果的融合方法的流程图,本实施例的方法具体包括如下步骤:
75.步骤310、训练得到至少两个目标检测模型;其中,不同目标检测模型的数据关注度不同。
76.步骤320、根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值。
77.步骤330、将待识别图像分别输入至各目标检测模型中,获取各目标检测模型输出的目标检测结果。
78.在本实施例中,所述目标检测结果包括:至少一个目标框的框描述信息,框描述信息包括:目标框的位置坐标、分类类型以及识别置信度。
79.步骤340、在各所述目标检测结果中,依次获取一个目标框作为当前处理目标框,并获取当前处理目标框的目标分类类型。
80.步骤350、获取与目标分类类型对应的目标重叠度阈值以及目标识别权重。
81.在此步骤中,所述目标识别权重为目标框对应的目标检测模型,对目标分类类型的识别权重。
82.步骤360、在各目标检测结果中,获取与当前处理目标框的重叠度大于或者等于目标重叠度阈值,且属于目标分类类型的各备选融合目标框。
83.在本实施例中,获取一个目标框作为当前处理目标框后,可以在各目标检测结果中包括的剩余目标框中,筛选属于目标分类类型的目标框,然后判断当前处理目标框与各筛选得到的目标框之间的重叠度是否大于或者等于目标重叠度阈值,若是,则将筛选得到的目标框作为备选融合目标框。
84.步骤370、根据当前处理目标框与各备选融合目标框的位置坐标、识别置信度以及目标识别权重,将当前处理目标框与各备选融合目标框进行融合,得到融合框的位置坐标以及识别置信度。
85.在本实施例中,可选的,可以对当前处理目标框与各备选融合目标框的位置坐标取平均值,得到融合框的位置坐标。其中,如果当前处理目标框的目标识别权重大于备选融合目标框的目标识别权重,则可以将当前处理目标框的识别置信度作为融合框的识别置信度。
86.这样设置的好处在于,通过当前处理目标框与各备选融合目标框的位置坐标,确定融合框的位置坐标,并通过当前处理目标框与各备选融合目标框的识别置信度以及目标识别权重,确定融合框的识别置信度,可以保证融合检测结果的位置和置信度更加准确。
87.在本实施例的一个实施方式中,根据当前处理目标框与各备选融合目标框的位置坐标、识别置信度以及目标识别权重,将当前处理目标框与各备选融合目标框进行融合,得到融合框的位置坐标以及识别置信度,包括:根据当前处理目标框与各备选融合目标框的位置坐标以及识别置信度,计算得到融合框的位置坐标;根据当前处理目标框与各备选融合目标框的目标识别权重以及识别置信度,得到融合框的识别置信度。
88.其中,假设当前处理目标框左上角的位置坐标为(ax1,ay1),右下角的位置坐标为(ax2,ay2),识别置信度为a,备选融合目标框左上角的位置坐标为(bx1,by1),右下角的位置坐标为(bx2,by2),识别置信度为b,将当前处理目标框与该备选融合目标框进行融合后,得到融合框左上角的位置坐标为(cx1,cy1),右下角的位置坐标为(cx2,cy2)。其中:
[0089][0090][0091]
在本实施例中,假设当前处理目标框的目标识别权重为w1,识别置信度为a,备选融合目标框的目标识别权重为w2,识别置信度为b,计算得到的融合框的识别置信度为c,则有:c=w1×
a+w2×
b。
[0092]
这样设置的好处在于,结合当前处理目标框与各备选融合目标框的位置坐标以及识别置信度,计算融合框的位置坐标,可以提高融合框位置计算结果的准确性;其次,根据当前处理目标框与各备选融合目标框的目标识别权重以及识别置信度,得到融合框的识别置信度,可以保证融合框的识别置信度接近于识别权重较高时目标框的识别置信度,进而可以提高融合检测结果的准确性。
[0093]
步骤380、判断是否完成对全部目标框的处理,若是,则结束所述目标检测结果的融合方法的执行,若否,则返回执行步骤340中在各目标检测结果中,依次获取一个目标框作为当前处理目标框的操作,直至完成对全部目标框的处理。
[0094]
本公开实施例通过训练得到至少两个目标检测模型,根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值,然后将待识别图像分别输入至各目标检测模型中,获取目标检测结果,并在各目标检测结果中,依次获取目标框作为当前处理目标框,并获取各备选融合目标框,根据当前处理目标框与各备选融合目标框的位置坐标、识别置信度以及目标识别权重,将当前处理目标框与各备选融合目标框进行融合,得到融合框的位置坐标以及识别置信度的技术手段,可以避免目标对象被漏检,提高目标检测结果的准确性。
[0095]
本公开实施例还提供了一种目标检测结果的融合装置,用于执行上述的目标检测结果的融合方法。
[0096]
图4为本公开实施例提供的一种目标检测结果的融合装置400的结构图,该装置包括:模型训练模块410、权重获取模块420、检测结果获取模块430和结果融合模块440。
[0097]
其中,模型训练模块410,用于训练得到至少两个目标检测模型;其中,不同目标检测模型的数据关注度不同;
[0098]
权重获取模块420,用于根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值;
[0099]
检测结果获取模块430,用于将待识别图像分别输入至各目标检测模型中,获取各目标检测模型输出的目标检测结果;
[0100]
结果融合模块440,用于根据所述识别权重以及所述重叠度阈值,对各目标检测结果进行融合处理,得到与待识别图像对应的融合检测结果。
[0101]
本公开实施例的技术方案,通过训练得到至少两个目标检测模型,根据训练得到的各目标检测模型,获取各目标检测模型针对不同分类类型的识别权重,以及与各分类类型分别对应的重叠度阈值,然后将待识别图像分别输入至各目标检测模型中,并获取各目标检测模型输出的目标检测结果,最后根据识别权重以及重叠度阈值,对各目标检测结果
进行融合处理,得到与待识别图像对应的融合检测结果的技术手段,可以避免目标对象被漏检,提高目标检测结果的准确性。
[0102]
在上述各实施例的基础上,模型训练模块410,包括:
[0103]
同一模型训练单元,用于使用不同数据关注度的训练样本图像集,训练同一模型结构的机器学习模型,得到各所述目标检测模型;
[0104]
不同模型训练单元,用于使用不同数据关注度的训练样本图像集,分别训练不同模型结构的机器学习模型,得到各所述目标检测模型,其中,数据关注度与设定分类类型关联。
[0105]
权重获取模块420,包括:
[0106]
测试样本输入单元,用于将测试样本图像集输入至各目标检测模型中,获取各所述目标检测模型针对各测试样本图像输出的比对检测结果;
[0107]
其中,测试样本图像中包括:至少一个标注框,以及与标注框对应的分类类型;
[0108]
结果比对单元,用于将各所述比对检测结果分别与测试样本图像集进行比对,获取各目标检测模型针对不同分类类型的识别准确度;
[0109]
权重确定单元,用于根据各目标检测模型针对各分类类型的识别准确度,确定各目标检测模型针对不同分类类型的识别权重;
[0110]
模型重叠度阈值筛选单元,用于在训练得到的各目标检测模型使用的模型参数中,筛选得到非极大值抑制nms算法所使用的模型重叠度阈值;
[0111]
重叠度阈值确定单元,用于根据各目标检测模型的模型重叠度阈值,以及目标检测模型的数据关注度与分类类型之间的关联关系,确定与各分类类型分别对应的重叠度阈值。
[0112]
其中,所述目标检测结果包括:至少一个目标框的框描述信息,框描述信息包括:目标框的位置坐标、分类类型以及识别置信度。
[0113]
结果融合模块440,包括:
[0114]
类型获取单元,用于在各所述目标检测结果中,依次获取一个目标框作为当前处理目标框,并获取当前处理目标框的目标分类类型;
[0115]
目标权重获取单元,用于获取与目标分类类型对应的目标重叠度阈值以及目标识别权重;
[0116]
备选融合目标框获取单元,用于在各目标检测结果中,获取与当前处理目标框的重叠度大于或者等于目标重叠度阈值,且属于目标分类类型的各备选融合目标框;
[0117]
目标框融合单元,用于根据当前处理目标框与各备选融合目标框的位置坐标、识别置信度以及目标识别权重,将当前处理目标框与各备选融合目标框进行融合,得到融合框的位置坐标以及识别置信度;
[0118]
全部目标框处理单元,用于返回执行在各所述目标检测结果中,依次获取一个目标框作为当前处理目标框的操作,直至完成对全部目标框的处理;
[0119]
目标框融合单元,包括:
[0120]
位置坐标计算子单元,用于根据当前处理目标框与各备选融合目标框的位置坐标以及识别置信度,计算得到融合框的位置坐标;
[0121]
置信度计算子单元,用于根据当前处理目标框与各备选融合目标框的目标识别权
重以及识别置信度,得到融合框的识别置信度。
[0122]
本公开实施例所提供的目标检测结果的融合装置可执行本公开任意实施例所提供的目标检测结果的融合方法,具备执行方法相应的功能模块和有益效果。
[0123]
根据本公开的实施例,本公开还提供了一种电子设备、一种可读存储介质和一种计算机程序产品。
[0124]
图5示出了可以用来实施本公开的实施例的示例电子设备500的示意性框图。电子设备旨在表示各种形式的数字计算机,诸如,膝上型计算机、台式计算机、工作台、个人数字助理、服务器、刀片式服务器、大型计算机、和其它适合的计算机。电子设备还可以表示各种形式的移动装置,诸如,个人数字处理、蜂窝电话、智能电话、可穿戴设备和其它类似的计算装置。本文所示的部件、它们的连接和关系、以及它们的功能仅仅作为示例,并且不意在限制本文中描述的和/或者要求的本公开的实现。
[0125]
如图5所示,设备500包括计算单元501,其可以根据存储在只读存储器(rom)502中的计算机程序或者从存储单元508加载到随机访问存储器(ram)503中的计算机程序,来执行各种适当的动作和处理。在ram 503中,还可存储设备500操作所需的各种程序和数据。计算单元501、rom 502以及ram 503通过总线504彼此相连。输入/输出(i/o)接口505也连接至总线504。
[0126]
设备500中的多个部件连接至i/o接口505,包括:输入单元506,例如键盘、鼠标等;输出单元507,例如各种类型的显示器、扬声器等;存储单元508,例如磁盘、光盘等;以及通信单元509,例如网卡、调制解调器、无线通信收发机等。通信单元509允许设备500通过诸如因特网的计算机网络和/或各种电信网络与其他设备交换信息/数据。
[0127]
计算单元501可以是各种具有处理和计算能力的通用和/或专用处理组件。计算单元501的一些示例包括但不限于中央处理单元(cpu)、图形处理单元(gpu)、各种专用的人工智能(ai)计算芯片、各种运行机器学习模型算法的计算单元、数字信号处理器(dsp)、以及任何适当的处理器、控制器、微控制器等。计算单元501执行上文所描述的各个方法和处理,例如目标检测结果的融合方法。例如,在一些实施例中,目标检测结果的融合方法可被实现为计算机软件程序,其被有形地包含于机器可读介质,例如存储单元508。在一些实施例中,计算机程序的部分或者全部可以经由rom 502和/或通信单元509而被载入和/或安装到设备500上。当计算机程序加载到ram 503并由计算单元501执行时,可以执行上文描述的目标检测结果的融合方法的一个或多个步骤。备选地,在其他实施例中,计算单元501可以通过其他任何适当的方式(例如,借助于固件)而被配置为执行目标检测结果的融合方法。
[0128]
本文中以上描述的系统和技术的各种实施方式可以在数字电子电路系统、集成电路系统、场可编程门阵列(fpga)、专用集成电路(asic)、专用标准产品(assp)、芯片上系统的系统(soc)、负载可编程逻辑设备(cpld)、计算机硬件、固件、软件、和/或它们的组合中实现。这些各种实施方式可以包括:实施在一个或者多个计算机程序中,该一个或者多个计算机程序可在包括至少一个可编程处理器的可编程系统上执行和/或解释,该可编程处理器可以是专用或者通用可编程处理器,可以从存储系统、至少一个输入装置、和至少一个输出装置接收数据和指令,并且将数据和指令传输至该存储系统、该至少一个输入装置、和该至少一个输出装置。
[0129]
用于实施本公开的方法的程序代码可以采用一个或多个编程语言的任何组合来
编写。这些程序代码可以提供给通用计算机、专用计算机或其他可编程数据处理装置的处理器或控制器,使得程序代码当由处理器或控制器执行时使流程图和/或框图中所规定的功能/操作被实施。程序代码可以完全在机器上执行、部分地在机器上执行,作为独立软件包部分地在机器上执行且部分地在远程机器上执行或完全在远程机器或服务器上执行。
[0130]
在本公开的上下文中,机器可读介质可以是有形的介质,其可以包含或存储以供指令执行系统、装置或设备使用或与指令执行系统、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体系统、装置或设备,或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(ram)、只读存储器(rom)、可擦除可编程只读存储器(eprom或快闪存储器)、光纤、便捷式紧凑盘只读存储器(cd

rom)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
[0131]
为了提供与用户的交互,可以在计算机上实施此处描述的系统和技术,该计算机具有:用于向用户显示信息的显示装置(例如,crt(阴极射线管)或者lcd(液晶显示器)监视器);以及键盘和指向装置(例如,鼠标或者轨迹球),用户可以通过该键盘和该指向装置来将输入提供给计算机。其它种类的装置还可以用于提供与用户的交互;例如,提供给用户的反馈可以是任何形式的传感反馈(例如,视觉反馈、听觉反馈、或者触觉反馈);并且可以用任何形式(包括声输入、语音输入或者、触觉输入)来接收来自用户的输入。
[0132]
可以将此处描述的系统和技术实施在包括后台部件的计算系统(例如,作为数据服务器)、或者包括中间件部件的计算系统(例如,应用服务器)、或者包括前端部件的计算系统(例如,具有图形用户界面或者网络浏览器的用户计算机,用户可以通过该图形用户界面或者该网络浏览器来与此处描述的系统和技术的实施方式交互)、或者包括这种后台部件、中间件部件、或者前端部件的任何组合的计算系统中。可以通过任何形式或者介质的数字数据通信(例如,通信网络)来将系统的部件相互连接。通信网络的示例包括:局域网(lan)、广域网(wan)、区块链网络和互联网。
[0133]
计算机系统可以包括客户端和服务器。客户端和服务器一般远离彼此并且通常通过通信网络进行交互。通过在相应的计算机上运行并且彼此具有客户端

服务器关系的计算机程序来产生客户端和服务器的关系。服务器可以是云服务器,又称为云计算服务器或云主机,是云计算服务体系中的一项主机产品,以解决了传统物理主机与vps服务中,存在的管理难度大,业务扩展性弱的缺陷。
[0134]
应该理解,可以使用上面所示的各种形式的流程,重新排序、增加或删除步骤。例如,本公开中记载的各步骤可以并行地执行也可以顺序地执行也可以不同的次序执行,只要能够实现本公开提供的技术方案所期望的结果,本文在此不进行限制。
[0135]
上述具体实施方式,并不构成对本公开保护范围的限制。本领域技术人员应该明白的是,根据设计要求和其他因素,可以进行各种修改、组合、子组合和替代。任何在本公开的精神和原则之内所作的修改、等同替换和改进等,均应包含在本公开保护范围之内。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1