一种图像处理方法、设备、服务器及存储介质转让专利

申请号 : CN202010949537.8

文献号 : CN111814776B

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 刘彦宏王洪斌

申请人 : 平安国际智慧城市科技股份有限公司

摘要 :

本发明实施例公开了一种图像处理方法、设备、服务器及存储介质,该方法包括:获取待处理的目标图像;将目标图像输入目标检测模型进行检测,以从目标图像中识别出的目标对象对应的目标检测框和目标类别;将目标图像输入鲁棒特征提取模型,以提取目标图像中目标对象的各组成部分对应的第一特征掩码集合;根据预设的特征掩码与类别的对应关系确定与目标类别对应的第二特征掩码集合,并计算第一特征掩码集合与第二特征掩码集合之间的相似度系数;根据相似度系数对目标图像中目标对象的目标类别进行矫正处理。通过这种基于鲁棒特征对受到攻击的目标图像中的目标对象进行分类校正的方法,增加了对抗攻击破解的难度,有效提高了图像处理的效率和准确率。

权利要求 :

1.一种图像处理方法,其特征在于,包括:

获取待处理的目标图像,所述目标图像中包括目标对象;

将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别;

将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合,所述第一特征掩码集合中包括一个或多个特征掩码,所述特征掩码是由数字组成,用于指示所述目标对象的各组成部分的鲁棒特征;

所述将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合,包括:将所述目标图像输入鲁棒特征提取模型,以确定所述目标图像中所述目标对象的各组成部分的像素覆盖区域;

提取所述目标对象的各组成部分的像素覆盖区域对应的所述第一特征掩码集合;

根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数,所述第二特征掩码集合中包括一个或多个所述特征掩码;

根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。

2.根据权利要求1所述的方法,其特征在于,所述将所述目标图像输入目标检测模型进行检测之前,包括:获取样本图像集,并确定所述样本图像集中各样本图像中的目标对象;

对所述各样本图像中的目标对象添加第一类别标签和检测框;

将所述添加第一类别标签和检测框的各样本图像输入深度神经网络模型进行训练,得到所述目标检测模型。

3.根据权利要求2所述的方法,其特征在于,所述将所述目标图像输入鲁棒特征提取模型之前,还包括:确定所述各样本图像中目标对象的组成部分;

对所述各样本图像中目标对象的各组成部分添加第二类别标签和特征掩码;

将所述添加第二类别标签和特征掩码的各样本图像输入所述深度神经网络模型训练,得到所述鲁棒特征提取模型。

4.根据权利要求1所述的方法,其特征在于,所述根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理,包括:检测所述相似度系数是否大于预设阈值;

如果检测结果为所述相似度系数大于预设阈值,则确定所述目标图像中的目标对象没有受到对抗攻击,不对所述目标对象的目标类型进行矫正处理;

如果检测结果为所述相似度系数小于或等于预设阈值,则确定所述目标图像中的所述目标对象受到对抗攻击,并对所述目标图像中的所述目标对象的目标类别进行矫正处理。

5.根据权利要求4所述的方法,其特征在于,所述对所述目标图像中的所述目标对象的目标类别进行矫正处理,包括:根据所述预设的特征掩码与类别的对应关系确定每个类别对应的特征掩码;

计算所述第一特征掩码集合与所述每个类别对应的特征掩码的相似度系数;

确定最大相似度系数对应的类别为所述目标对象的目标类别。

6.根据权利要求1所述的方法,其特征在于,所述计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数,包括:获取所述第一特征掩码集合与所述第二特征掩码集合的交集特征掩码;

获取所述第一特征掩码集合与所述第二特征掩码集合的并集特征掩码;

根据所述交集特征掩码与所述并集特征掩码的比值的绝对值,确定所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。

7.一种图像处理设备,其特征在于,包括:

获取单元,用于获取待处理的目标图像,所述目标图像中包括目标对象;

检测单元,用于将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别;

提取单元,用于将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合,所述第一特征掩码集合中包括一个或多个特征掩码,所述特征掩码是由数字组成,用于指示所述目标对象的各组成部分的鲁棒特征;

所述提取单元将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合时,具体用于将所述目标图像输入鲁棒特征提取模型,以确定所述目标图像中所述目标对象的各组成部分的像素覆盖区域;提取所述目标对象的各组成部分的像素覆盖区域对应的所述第一特征掩码集合;

确定单元,用于根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数,所述第二特征掩码集合中包括一个或多个所述特征掩码;

矫正单元,用于根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。

8.一种服务器,其特征在于,包括处理器、输入设备、输出设备和存储器,所述处理器、输入设备、输出设备和存储器相互连接,其中,所述存储器用于存储计算机程序,所述计算机程序包括程序,所述处理器被配置用于调用所述程序,执行如权利要求1-6任一项所述的方法。

9.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行以实现权利要求1-6任一项所述的方法。

说明书 :

一种图像处理方法、设备、服务器及存储介质

技术领域

[0001] 本发明涉及图像处理技术领域,尤其涉及一种图像处理方法、设备、服务器及存储介质。

背景技术

[0002] 智慧城市建设中会涉及到很多使用智能监控摄像头的应用,对场景中拍摄得到的监控图像中的不同类型的目标对象进行检测识别。例如,在社区安防,食品监管,环境监管,交通监控等应用中,对特定的人、物进行检测,并且,在安防和监管等应用中对检测的鲁棒性要求较高。目前,人们通过深度卷积神经网络技术来实现监控图像中目标对象的检测与识别,目标检测技术通过在预先定义的图像数据集上进行训练获得模型,然后在实际场景中用该模型对在线实时获取的图像进行预测。
[0003] 然而,深度神经网络模型在处理受到对抗攻击的图像时,准确率急剧下降,基于范数模型的对抗防御方法仅在像素的扰动值小于某个阈值时能提供鲁棒性,对于扰动范围大于阈值的攻击没有提供有效的解决方案。因此,如何提高图像处理过程中的鲁棒性是非常重要的。

发明内容

[0004] 本发明实施例提供了一种图像处理方法、设备、服务器及存储介质,通过基于鲁棒特征对受到攻击的目标图像中的目标对象进行分类校正,增加了对抗攻击破解的难度,攻击者不仅需要改变模型的预测类别,还需要改变各个鲁棒特征,同时,对于未受到攻击的目标图像可以继续使用原有深度神经网络模型进行预测,利用非鲁棒特征保持预测的高准确率,有效地提高了图像处理的效率和准确率。
[0005] 第一方面,本发明实施例提供了一种图像处理方法,其特征在于,包括:
[0006] 获取待处理的目标图像,所述目标图像中包括目标对象;
[0007] 将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别;
[0008] 将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合;
[0009] 根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数;
[0010] 根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。
[0011] 进一步地,所述将所述目标图像输入目标检测模型进行检测之前,还包括:
[0012] 获取样本图像集,并确定所述样本图像集中各样本图像中的目标对象;
[0013] 对所述各样本图像中的目标对象添加第一类别标签和检测框;
[0014] 将所述添加第一类别标签和检测框的各样本图像输入深度神经网络模型进行训练,得到所述目标检测模型。
[0015] 进一步地,所述将所述目标图像输入鲁棒特征提取模型之前,还包括:
[0016] 确定所述各样本图像中目标对象的组成部分;
[0017] 对所述各样本图像中目标对象的各组成部分添加第二类别标签和特征掩码;
[0018] 将所述添加第二类别标签和特征掩码的各样本图像输入所述深度神经网络模型训练,得到所述鲁棒特征提取模型。
[0019] 进一步地,所述将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合,包括:
[0020] 将所述目标图像输入鲁棒特征提取模型,以确定所述目标图像中所述目标对象的各组成部分的像素覆盖区域;
[0021] 提取所述目标对象的各组成部分的像素覆盖区域对应的所述第一特征掩码集合。
[0022] 进一步地,所述根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理,包括:
[0023] 检测所述相似度系数是否大于预设阈值;
[0024] 如果检测结果为所述相似度系数大于预设阈值,则确定所述目标图像中的目标对象没有受到对抗攻击,不对所述目标对象的目标类型进行矫正处理;
[0025] 如果检测结果为所述相似度系数小于或等于预设阈值,则确定所述目标图像中的所述目标对象受到对抗攻击,并对所述目标图像中的所述目标对象的目标类别进行矫正处理。
[0026] 进一步地,所述对所述目标图像中的所述目标对象的目标类别进行矫正处理,包括:
[0027] 根据所述预设的特征掩码与类别的对应关系确定每个类别对应的特征掩码;
[0028] 计算所述第一特征掩码集合与所述每个类别对应的特征掩码的相似度系数;
[0029] 确定最大相似度系数对应的类别为所述目标对象的目标类别。
[0030] 进一步地,所述计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数,包括:
[0031] 获取所述第一特征掩码集合与所述第二特征掩码集合的交集特征掩码;
[0032] 获取所述第一特征掩码集合与所述第二特征掩码集合的并集特征掩码;
[0033] 根据所述交集特征掩码与所述并集特征掩码的比值的绝对值,确定所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。
[0034] 第二方面,本发明实施例提供了一种图像处理设备,其特征在于,包括:
[0035] 获取待处理的目标图像,所述目标图像中包括目标对象;
[0036] 将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别;
[0037] 将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合;
[0038] 根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数;
[0039] 根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。
[0040] 第三方面,本发明实施例提供了一种服务器,包括处理器、输入设备、输出设备和存储器,所述处理器、输入设备、输出设备和存储器相互连接,其中,所述存储器用于存储支持图像处理设备执行上述方法的计算机程序,所述计算机程序包括程序,所述处理器被配置用于调用所述程序,执行上述第一方面的方法。
[0041] 第四方面,本发明实施例提供了一种计算机可读存储介质,所述计算机存储介质存储有计算机程序,所述计算机程序被处理器执行以实现上述第一方面的方法。
[0042] 本发明实施例中,服务器可以获取待处理的目标图像,并将待处理的目标图像输入目标检测模型进行检测,以从目标图像中识别出的目标对象对应的目标检测框和目标类别,将目标图像输入鲁棒特征提取模型,以提取目标图像中目标对象的各组成部分对应的第一特征掩码集合,以及根据预设的特征掩码与类别的对应关系确定与目标类别对应的第二特征掩码集合,并计算第一特征掩码集合与第二特征掩码集合之间的相似度系数,从而根据相似度系数对目标图像中目标对象的目标类别进行矫正处理。通过这种基于鲁棒特征对受到攻击的目标图像中的目标对象进行分类校正的方式,增加了对抗攻击破解的难度,攻击者不仅需要改变模型的预测类别,还需要改变各个鲁棒特征,同时,对于未受到攻击的目标图像可以继续使用原有深度神经网络模型进行预测,利用非鲁棒特征保持预测的高准确率,有效地提高了图像处理的效率和准确率。

附图说明

[0043] 为了更清楚地说明本发明实施例技术方案,下面将对实施例描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
[0044] 图1是本发明实施例提供的一种图像处理方法的示意流程图;
[0045] 图2是本发明实施例提供的一种图像处理设备的示意框图;
[0046] 图3是本发明实施例提供的一种服务器的示意框图。

具体实施方式

[0047] 下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
[0048] 本发明实施例提供的一种图像处理方法可以应用于一种图像处理设备,所述图像处理设备可以设置在服务器中。
[0049] 下面结合附图1对本发明实施例提供的图像处理方法进行示意性说明。
[0050] 请参见图1,图1是本发明实施例提供的一种图像处理方法的示意流程图,如图1所示,该方法可以由图像处理设备执行,所述图像处理设备设置于服务器中。具体地,本发明实施例的所述方法包括如下步骤。
[0051] S101:获取待处理的目标图像,所述目标图像中包括目标对象。
[0052] 本发明实施例中,图像处理设备可以获取待处理的目标图像,所述目标图像中包括目标对象。在某些实施例中,所述目标图像中包括一个或多个目标对象,所述目标对象可以是人、物等任一物体。
[0053] 在某些实施例中,所述目标图像可以是拍摄装置拍摄得到的;在某些实施例中,所述拍摄装置可以包括但不限于相机、传感器等,所述拍摄装置可以用于监控场景。在某些实施例中,所述图像处理设备可以与拍摄装置建立通信连接,所述图像处理设备可以获取拍摄装置拍摄到的目标图像。
[0054] S102:将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别。
[0055] 本发明实施例中,图像处理设备可以将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别。
[0056] 在一个实施例中,图像处理设备在将所述目标图像输入目标检测模型进行检测之前,可以获取样本图像集,并确定所述样本图像集中各样本图像中的目标对象,以及对所述各样本图像中的目标对象添加第一类别标签和检测框,以及将所述添加第一类别标签和检测框的各样本图像输入深度神经网络模型进行训练,得到所述目标检测模型。在某些实施例中,所述第一类别标签用于指示各样本图像中各目标对象的类别。在某些实施例中,所述检测框可以通过线条组成的闭合框,其中,所述线条组成的闭合框可以是任意形状,在一个示例中,所述线条组成的闭合框可以是圆形框、方形框、多边形框、不规则形状框等,此处不做具体限定。
[0057] S103:将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合。
[0058] 本发明实施例中,图像处理设备可以将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合。在某些实施例中,所述第一特征掩码集合中包括一个或多个特征掩码。
[0059] 在一个实施例中,图像处理设备在将所述目标图像输入鲁棒特征提取模型之前,可以确定所述各样本图像中目标对象的组成部分,并对所述各样本图像中目标对象的各组成部分添加第二类别标签和特征掩码,以及将所述添加第二类别标签和特征掩码的各样本图像输入所述深度神经网络模型训练,得到所述鲁棒特征提取模型。在某些实施例中,所述特征掩码是由数字组成,用于指示所述目标对象的各组成部分的鲁棒特征。
[0060] 在一个示例中,假设样本图像中的目标对象为汽车,则汽车的组成部分包括轮胎、车窗、车架、雨刮器等。
[0061] 在一个实施例中,图像处理设备在将所述目标图像输入鲁棒特征提取模型之前,可以从样本图像集中提取出部分包括目标对象的子样本图像,并确定各子样本图像中目标对象的组成部分,并对所述各子样本图像中目标对象的各组成部分添加第二类别标签和第一特征掩码,以及将所述添加第二类别标签和第一特征掩码的各子样本图像输入所述深度神经网络模型训练,得到所述鲁棒特征提取模型。
[0062] 在一个实施例中,图像处理设备在将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合时,可以将所述目标图像输入鲁棒特征提取模型,以确定所述目标图像中所述目标对象的各组成部分的像素覆盖区域,并提取所述目标对象的各组成部分的像素覆盖区域对应的所述第一特征掩码集合。
[0063] 在一个示例中,假设目标对象为汽车,汽车的组成部分包括车窗,则车窗可以用不同的颜色对应的第一特征掩码来表示车窗的像素覆盖区域。
[0064] S104:根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。
[0065] 本发明实施例中,图像处理设备可以根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。在某些实施例中,所述第二特征掩码集合中包括一个或多个特征掩码。
[0066] 在一个实施例中,图像处理设备在计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数时,可以获取所述第一特征掩码集合与所述第二特征掩码集合的交集特征掩码,并获取所述第一特征掩码集合与所述第二特征掩码集合的并集特征掩码,以及根据所述交集特征掩码与所述并集特征掩码的比值的绝对值,确定所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。
[0067] 在某些实施例中,所述预设的特征掩码与类别的对应关系可以通过矩阵的形式来表示,所述矩阵是根据预设的特征掩码与类别建立的。
[0068] 在一个示例中,假设第一特征掩码集合为fri,第二特征掩码集合为fei,则可以根据如下公式(1)计算第一特征掩码集合fri与第二特征掩码集合fei之间的相似度系数J(fri,fei)。
[0069]
[0070] S105:根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。
[0071] 本发明实施例中,图像处理设备可以根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。
[0072] 在一个实施例中,图像处理设备在根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理时,可以检测所述相似度系数是否大于预设阈值,如果检测结果为所述相似度系数大于预设阈值,则可以确定所述目标图像中的目标对象没有受到对抗攻击,不对所述目标对象的目标类型进行矫正处理,如果检测结果为所述相似度系数小于或等于预设阈值,则可以确定所述目标图像中的所述目标对象受到对抗攻击,并对所述目标图像中的所述目标对象的目标类别进行矫正处理。
[0073] 例如,如果图像处理设备检测到相似度系数J大于预设阈值t,则可以确定所述目标图像中的目标对象bi没有受到对抗攻击,不对所述目标对象的目标类型进行矫正处理,并确定当前识别出的目标对象的第一标签为目标对象最终的目标类别;如果检测到相似度系数J小于或等于预设阈值t,则可以确定所述目标图像中的所述目标对象受到对抗攻击,需要对所述目标图像中的所述目标对象的目标类别进行矫正处理。
[0074] 在一个实施例中,图像处理设备在对所述目标图像中的所述目标对象的目标类别进行矫正处理时,可以根据所述预设的特征掩码与类别的对应关系确定每个类别对应的特征掩码,并计算所述第一特征掩码集合与所述每个类别对应的特征掩码的相似度系数,以及确定最大相似度系数对应的类别为所述目标对象的目标类别。
[0075] 在一个实施例中,图像处理设备在确定最大相似度系数对应的类别为所述目标对象的目标类别时,可以获取最大相似度系数对应的第一类别标签,并将该第一类别标签添加到所述目标对象中,以确定所述最大相似度系数对应的类别为所述目标对象的目标类别。
[0076] 例如,假设根据预设的特征掩码与类别的对应关系确定每个类别对应的特征掩码,并计算所述第一特征掩码集合与所述每个类别对应的特征掩码的相似度系数,以及确定最大相似度系数对应的第一类别标签cj对应的类别为所述目标对象的目标类别。
[0077] 本发明实施例中,图像处理设备可以获取待处理的目标图像,并将待处理的目标图像输入目标检测模型进行检测,以从目标图像中识别出的目标对象对应的目标检测框和目标类别,将目标图像输入鲁棒特征提取模型,以提取目标图像中目标对象的各组成部分对应的第一特征掩码集合,以及根据预设的特征掩码与类别的对应关系确定与目标类别对应的第二特征掩码集合,并计算第一特征掩码集合与第二特征掩码集合之间的相似度系数,从而根据相似度系数对目标图像中目标对象的目标类别进行矫正处理。通过这种基于鲁棒特征对受到攻击的目标图像中的目标对象进行分类校正的方式,增加了对抗攻击破解的难度,攻击者不仅需要改变模型的预测类别,还需要改变各个鲁棒特征,同时,对于未受到攻击的目标图像可以继续使用原有深度神经网络模型进行预测,利用非鲁棒特征保持预测的高准确率,有效地提高了图像处理的效率和准确率。
[0078] 本发明实施例还提供了一种图像处理设备,该图像处理设备用于执行前述任一项所述的方法的单元。具体地,参见图2,图2是本发明实施例提供的一种图像处理设备的示意框图。本实施例的图像处理设备包括:获取单元201、检测单元202、提取单元203、确定单元204以及矫正单元205。
[0079] 获取单元201,用于获取待处理的目标图像,所述目标图像中包括目标对象;
[0080] 检测单元202,用于将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别;
[0081] 提取单元203,用于将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合;
[0082] 确定单元204,用于根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数;
[0083] 矫正单元205,用于根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。
[0084] 进一步地,所述检测单元202将所述目标图像输入目标检测模型进行检测之前,还用于:
[0085] 获取样本图像集,并确定所述样本图像集中各样本图像中的目标对象;
[0086] 对所述各样本图像中的目标对象添加第一类别标签和检测框;
[0087] 将所述添加第一类别标签和检测框的各样本图像输入深度神经网络模型进行训练,得到所述目标检测模型。
[0088] 进一步地,所述提取单元203将所述目标图像输入鲁棒特征提取模型之前,还用于:
[0089] 确定所述各样本图像中目标对象的组成部分;
[0090] 对所述各样本图像中目标对象的各组成部分添加第二类别标签和特征掩码;
[0091] 将所述添加第二类别标签和特征掩码的各样本图像输入所述深度神经网络模型训练,得到所述鲁棒特征提取模型。
[0092] 进一步地,所述提取单元203将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合时,具体用于:
[0093] 将所述目标图像输入鲁棒特征提取模型,以确定所述目标图像中所述目标对象的各组成部分的像素覆盖区域;
[0094] 提取所述目标对象的各组成部分的像素覆盖区域对应的所述第一特征掩码集合。
[0095] 进一步地,所述矫正单元205根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理时,具体用于:
[0096] 检测所述相似度系数是否大于预设阈值;
[0097] 如果检测结果为所述相似度系数大于预设阈值,则确定所述目标图像中的目标对象没有受到对抗攻击,不对所述目标对象的目标类型进行矫正处理;
[0098] 如果检测结果为所述相似度系数小于或等于预设阈值,则确定所述目标图像中的所述目标对象受到对抗攻击,并对所述目标图像中的所述目标对象的目标类别进行矫正处理。
[0099] 进一步地,所述矫正单元205对所述目标图像中的所述目标对象的目标类别进行矫正处理时,具体用于:
[0100] 根据所述预设的特征掩码与类别的对应关系确定每个类别对应的特征掩码;
[0101] 计算所述第一特征掩码集合与所述每个类别对应的特征掩码的相似度系数;
[0102] 确定最大相似度系数对应的类别为所述目标对象的目标类别。
[0103] 进一步地,所述确定单元204计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数时,具体用于:
[0104] 获取所述第一特征掩码集合与所述第二特征掩码集合的交集特征掩码;
[0105] 获取所述第一特征掩码集合与所述第二特征掩码集合的并集特征掩码;
[0106] 根据所述交集特征掩码与所述并集特征掩码的比值的绝对值,确定所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。
[0107] 本发明实施例中,图像处理设备可以获取待处理的目标图像,并将待处理的目标图像输入目标检测模型进行检测,以从目标图像中识别出的目标对象对应的目标检测框和目标类别,将目标图像输入鲁棒特征提取模型,以提取目标图像中目标对象的各组成部分对应的第一特征掩码集合,以及根据预设的特征掩码与类别的对应关系确定与目标类别对应的第二特征掩码集合,并计算第一特征掩码集合与第二特征掩码集合之间的相似度系数,从而根据相似度系数对目标图像中目标对象的目标类别进行矫正处理。通过这种基于鲁棒特征对受到攻击的目标图像中的目标对象进行分类校正的方式,增加了对抗攻击破解的难度,攻击者不仅需要改变模型的预测类别,还需要改变各个鲁棒特征,同时,对于未受到攻击的目标图像可以继续使用原有深度神经网络模型进行预测,利用非鲁棒特征保持预测的高准确率,有效地提高了图像处理的效率和准确率。
[0108] 参见图3,图3是本发明实施例提供的一种服务器的示意框图。如图所示的本实施例中的服务器可以包括:一个或多个处理器301;一个或多个输入设备302,一个或多个输出设备303和存储器304。上述处理器301、输入设备302、输出设备303和存储器304通过总线305连接。存储器304用于存储计算机程序,所述计算机程序包括程序,处理器301 用于执行存储器304存储的程序。其中,处理器301被配置用于调用所述程序执行:
[0109] 获取待处理的目标图像,所述目标图像中包括目标对象;
[0110] 将所述目标图像输入目标检测模型进行检测,以从所述目标图像中识别出的所述目标对象对应的目标检测框和目标类别;
[0111] 将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合;
[0112] 根据预设的特征掩码与类别的对应关系确定与所述目标类别对应的第二特征掩码集合,并计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数;
[0113] 根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理。
[0114] 进一步地,所述处理器301将所述目标图像输入目标检测模型进行检测之前,还用于:
[0115] 获取样本图像集,并确定所述样本图像集中各样本图像中的目标对象;
[0116] 对所述各样本图像中的目标对象添加第一类别标签和检测框;
[0117] 将所述添加第一类别标签和检测框的各样本图像输入深度神经网络模型进行训练,得到所述目标检测模型。
[0118] 进一步地,所述处理器301将所述目标图像输入鲁棒特征提取模型之前,还用于:
[0119] 确定所述各样本图像中目标对象的组成部分;
[0120] 对所述各样本图像中目标对象的各组成部分添加第二类别标签和特征掩码;
[0121] 将所述添加第二类别标签和特征掩码的各样本图像输入所述深度神经网络模型训练,得到所述鲁棒特征提取模型。
[0122] 进一步地,所述处理器301将所述目标图像输入鲁棒特征提取模型,以提取所述目标图像中所述目标对象的各组成部分对应的第一特征掩码集合时,具体用于:
[0123] 将所述目标图像输入鲁棒特征提取模型,以确定所述目标图像中所述目标对象的各组成部分的像素覆盖区域;
[0124] 提取所述目标对象的各组成部分的像素覆盖区域对应的所述第一特征掩码集合。
[0125] 进一步地,所述处理器301根据所述相似度系数对所述目标图像中所述目标对象的目标类别进行矫正处理时,具体用于:
[0126] 检测所述相似度系数是否大于预设阈值;
[0127] 如果检测结果为所述相似度系数大于预设阈值,则确定所述目标图像中的目标对象没有受到对抗攻击,不对所述目标对象的目标类型进行矫正处理;
[0128] 如果检测结果为所述相似度系数小于或等于预设阈值,则确定所述目标图像中的所述目标对象受到对抗攻击,并对所述目标图像中的所述目标对象的目标类别进行矫正处理。
[0129] 进一步地,所述处理器301对所述目标图像中的所述目标对象的目标类别进行矫正处理时,具体用于:
[0130] 根据所述预设的特征掩码与类别的对应关系确定每个类别对应的特征掩码;
[0131] 计算所述第一特征掩码集合与所述每个类别对应的特征掩码的相似度系数;
[0132] 确定最大相似度系数对应的类别为所述目标对象的目标类别。
[0133] 进一步地,所述处理器301计算所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数时,具体用于:
[0134] 获取所述第一特征掩码集合与所述第二特征掩码集合的交集特征掩码;
[0135] 获取所述第一特征掩码集合与所述第二特征掩码集合的并集特征掩码;
[0136] 根据所述交集特征掩码与所述并集特征掩码的比值的绝对值,确定所述第一特征掩码集合与所述第二特征掩码集合之间的相似度系数。
[0137] 本发明实施例中,服务器可以获取待处理的目标图像,并将待处理的目标图像输入目标检测模型进行检测,以从目标图像中识别出的目标对象对应的目标检测框和目标类别,将目标图像输入鲁棒特征提取模型,以提取目标图像中目标对象的各组成部分对应的第一特征掩码集合,以及根据预设的特征掩码与类别的对应关系确定与目标类别对应的第二特征掩码集合,并计算第一特征掩码集合与第二特征掩码集合之间的相似度系数,从而根据相似度系数对目标图像中目标对象的目标类别进行矫正处理。通过这种基于鲁棒特征对受到攻击的目标图像中的目标对象进行分类校正的方式,增加了对抗攻击破解的难度,攻击者不仅需要改变模型的预测类别,还需要改变各个鲁棒特征,同时,对于未受到攻击的目标图像可以继续使用原有深度神经网络模型进行预测,利用非鲁棒特征保持预测的高准确率,有效地提高了图像处理的效率和准确率。
[0138] 应当理解,在本发明实施例中,所称处理器301可以是中央处理单元 (CenSral Processing UniS,CPU),该处理器还可以是其他通用处理器、数字信号处理器 (DigiSal Signal Processor,DSP)、专用集成电路 (ApplicaSion Specific InSegraSed CircuiS,ASIC)、现成可编程门阵列 (Field-Programmable GaSe Array,FPGA) 或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
[0139] 输入设备302可以包括触控板、麦克风等,输出设备303可以包括显示器(LCD等)、扬声器等。
[0140] 该存储器304可以包括只读存储器和随机存取存储器,并向处理器301 提供指令和数据。存储器304的一部分还可以包括非易失性随机存取存储器。例如,存储器304还可以存储设备类型的信息。
[0141] 具体实现中,本发明实施例中所描述的处理器301、输入设备302、输出设备303可执行本发明实施例提供的图2所述的方法实施例中所描述的实现方式,也可执行本发明实施例图3所描述的图像处理设备的实现方式,在此不再赘述。
[0142] 本发明实施例中还提供了一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现图2所对应实施例中描述的图像处理方法,也可实现本发明图3所对应实施例的图像处理设备,在此不再赘述。
[0143] 所述计算机可读存储介质可以是前述任一实施例所述的图像处理设备的内部存储单元,例如图像处理设备的硬盘或内存。所述计算机可读存储介质也可以是所述图像处理设备的外部存储设备,例如所述图像处理设备上配备的插接式硬盘,智能存储卡(SmarS Media Card, SMC),安全数字(Secure DigiSal, SD)卡,闪存卡(Flash Card)等。进一步地,所述计算机可读存储介质还可以既包括所述图像处理设备的内部存储单元也包括外部存储设备。所述计算机可读存储介质用于存储所述计算机程序以及所述图像处理设备所需的其他程序和数据。所述计算机可读存储介质还可以用于暂时地存储已经输出或者将要输出的数据。
[0144] 所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本发明的技术方案本质上或者说对现有技术做出贡献的部分,或者该技术方案的全部或部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个计算机可读存储介质中,包括若干指令用以使得一台计算机设备 ( 可以是个人计算机,终端,或者网络设备等 ) 执行本发明各个实施例所述方法的全部或部分步骤。而前述的计算机可读存储介质包括:U 盘、移动硬盘、只读存储器 (ROM,Read-Only Memory)、随机存取存储器 (RAM,Random Access Memory)、磁碟或者光盘等各种可以存储程序代码的介质。
[0145] 以上所述,仅为本发明的部分实施方式,但本发明的保护范围并不局限于此,任何熟悉本技术领域的技术人员在本发明揭露的技术范围内,可轻易想到各种等效的修改或替换,这些修改或替换都应涵盖在本发明的保护范围之内。