CN114612667A - 图像分割方法、车辆控制方法及装置 - Google Patents

图像分割方法、车辆控制方法及装置 Download PDF

Info

Publication number
CN114612667A
CN114612667A CN202210283973.5A CN202210283973A CN114612667A CN 114612667 A CN114612667 A CN 114612667A CN 202210283973 A CN202210283973 A CN 202210283973A CN 114612667 A CN114612667 A CN 114612667A
Authority
CN
China
Prior art keywords
query
image
panoramic
feature information
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
Application number
CN202210283973.5A
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.)
Shanghai Sensetime Lingang Intelligent Technology Co Ltd
Original Assignee
Shanghai Sensetime Lingang Intelligent Technology 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 Shanghai Sensetime Lingang Intelligent Technology Co Ltd filed Critical Shanghai Sensetime Lingang Intelligent Technology Co Ltd
Priority to CN202210283973.5A priority Critical patent/CN114612667A/zh
Publication of CN114612667A publication Critical patent/CN114612667A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/24Classification techniques
    • G06F18/241Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Data Mining & Analysis (AREA)
  • Evolutionary Computation (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Artificial Intelligence (AREA)
  • General Physics & Mathematics (AREA)
  • General Engineering & Computer Science (AREA)
  • General Health & Medical Sciences (AREA)
  • Software Systems (AREA)
  • Molecular Biology (AREA)
  • Computing Systems (AREA)
  • Biophysics (AREA)
  • Biomedical Technology (AREA)
  • Mathematical Physics (AREA)
  • Computational Linguistics (AREA)
  • Health & Medical Sciences (AREA)
  • Bioinformatics & Cheminformatics (AREA)
  • Bioinformatics & Computational Biology (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Evolutionary Biology (AREA)
  • Image Analysis (AREA)

Abstract

本公开提供了一种图像分割方法、车辆控制方法及装置,其中,图像分割方法包括:获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。

Description

图像分割方法、车辆控制方法及装置
技术领域
本公开涉及图像处理技术领域,具体而言,涉及一种图像分割方法、车辆控制方法及装置。
背景技术
图像分割是图像处理和是机器视觉技术中关于图像理解的重要一环。在进行图像分割时,可以通过对图像中每一个像素点进行分类,确定每个点的类别,从而实现区域划分。目前,常见的图像分割模型包括全景分割模型和部件分割模型,所述全景分割模型可以划分出待分割图像中的前景物体(比如人、汽车等),以及后景物体(比如马路、建筑等);所述部件分割模型可以将待分割图像中的可数物体的部件进行分割,比如人的头部、汽车的车灯等。
相关技术中,为了得到更精细化的分割效果,往往会选择使用全景分割模型和部件分割模型分别对待分割图像进行分割,并将全景分割结果和部件分割结果进行融合处理,虽然通过这种方式得到的融合后的分割结果,相较于只使用一个图像分割模型进行分割得到的分割结果的图像分割效果更好,但由于使用两个独立建模的模型进行图像分割,导致图像分割效率低下。
发明内容
本公开实施例至少提供一种图像分割、车辆控制方法、装置、计算机设备及存储介质。
第一方面,本公开实施例提供了一种图像分割方法,包括:
获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;
分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;
基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
这样,通过引入查询向量分别对,目标图像对应的初始特征图进行特征提取后得到的全景特征信息和部件特征信息进行特征查询,并基于特征查询后的第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。这样,通过基于查询向量在一个模型中分别执行全景图像分割任务和部件图像分割任务,相较于使用多个独立建模的模型进行图像分割,提高对所述目标图像的图像分割效果和图像分割效率。
一种可能的实施方式中,所述基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果,包括:
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图;以及,基于所述第一查询向量和所述第二查询向量确定全景查询向量;
基于所述全景特征图对所述全景查询向量进行调整,并基于调整后的全景查询向量重新确定第一查询向量和第二查询向量;
基于重新确定的第一查询向量重新确定第一查询特征信息,以及基于重新确定的第二查询向量,重新确定第二查询特征信息;
基于重新确定的第一查询特征信息、重新确定的第二查询特征信息、重新确定的第一查询向量和重新确定的第二查询向量,返回执行确定全景特征图和全景查询向量的步骤,直至满足预设条件;
将最后一次确定的第一查询特征信息作为第一目标查询特征信息,以及将最后一次确定的第二查询特征信息作为第二目标查询特征信息,并基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果。
一种可能的实施方式中,所述基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图,包括:
基于所述第一查询特征信息和所述全景特征信息,确定所述目标图像对应的第一特征图;以及,基于所述第二查询特征信息和所述部件特征信息,确定所述目标图像对应的第二特征图;
将所述第一特征图和所述第二特征图进行拼接,得到所述全景特征图。
这样,通过确定第一特征图和第二特征图,并对所述第一特征图和第二特征图进行拼接得到全景特征图,使得后续处理时只需要处理一个特征图即可,从而可以提高图像分割效率。
一种可能的实施方式中,所述基于所述第一查询向量和所述第二查询向量确定全景查询向量,包括:
将所述第一查询向量和所述第二查询向量进行拼接,得到所述全景查询向量。
这样,通过将第一查询向量和第二查询向量进行拼接得到全景查询向量,使得后续处理时只需要处理一个查询向量即可,从而可以提高图像分割效率。
一种可能的实施方式中,所述基于所述全景特征图对所述全景查询向量进行调整,包括:
基于所述全景查询向量和所述全景特征图进行动态卷积处理,得到中间查询向量;
基于自注意力机制算法对所述中间查询向量进行处理,得到调整后的全景查询向量。
这样,基于动态卷积处理和自注意力机制算法对所述全景查询向量进行调整,可以使得调整后的全景查询向量能够更好的提取出目标图像对应的全景特征(第一查询特征信息)和局部特征(第二查询特征信息),从而提高图像分割效果。
一种可能的实施方式中,所述基于调整后的全景查询向量重新确定第一查询向量和第二查询向量,包括:
对调整后的全景查询向量进行拆分,得到重新确定的第一查询向量和第二查询向量。
这样,通过对调整后的全景查询向量进行拆分,可以完成对所述第一查询向量和所述第二查询向量的一次更新,并可以通过后续的更新,使得查询向量能够更好的提取出目标图像对应的全景特征(第一查询特征信息)和局部特征(第二查询特征信息),从而提高图像分割效果。
一种可能的实施方式中,所述基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果,包括:
将所述第一目标查询特征信息和所述第二目标查询特征信息进行融合处理,得到所述目标图像的目标分割结果。
这样,通过对第一目标查询特征信息和第二目标查询特征信息进行融合处理,可以得到融合了全景分割结果和部件分割结果的目标分割结果。
一种可能的实施方式中,所述图像分割方法利用图像分割网络完成。
一种可能的实施方式中,所述图像分割网络采用以下步骤训练得到:
获取样本图像以及所述样本图像对应的样本全景分割图像和样本部件分割图像;
将所述样本图像输入至待训练的图像分割网络,确定与所述样本图像对应的第一样本查询特征信息和第二样本查询特征信息;
基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,并基于所述目标损失值对所述待训练的图像分割网络的网络参数进行调整,所述网络参数包括第一查询向量和第二查询向量。
一种可能的实施方式中,所述基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,包括:
基于所述第一样本查询特征信息和所述样本全景分割图像,确定全景分割损失;以及,
基于所述第二样本查询特征信息和所述样本部件分割图像,确定部件分割损失;以及,
基于所述第一样本查询特征信息和所述第二样本查询特征信息确定样本全景特征图,并基于样本全景特征图和所述样本图像对应的分类标签,确定分类损失;
基于所述全景分割损失、所述部件分割损失和所述分类损失,确定所述目标损失值。
这样,基于分割损失和分类损失确定目标损失值,可以使得后续在基于所述目标损失值对待训练的图像分割网络的网络参数进行调整后,图像分割网络的网络精度更高,从而提高图像分割效率。
第二方面,本公开实施例提供了一种车辆控制方法,包括:
获取自动驾驶车辆上部署的图像采集装置采集的待分割图像;
基于第一方面任一所述的图像分割方法,确定所述待分割图像对应的图像分割结果;
基于所述图像分割结果对所述自动驾驶车辆进行控制。
第三方面,本公开实施例还提供一种图像分割装置,包括:
第一获取模块,用于获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;
提取模块,用于分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;
查询模块,用于基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;
第一确定模块,用于基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
一种可能的实施方式中,所述第一确定模块,在基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果时,用于:
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图;以及,基于所述第一查询向量和所述第二查询向量确定全景查询向量;
基于所述全景特征图对所述全景查询向量进行调整,并基于调整后的全景查询向量重新确定第一查询向量和第二查询向量;
基于重新确定的第一查询向量重新确定第一查询特征信息,以及基于重新确定的第二查询向量,重新确定第二查询特征信息;
基于重新确定的第一查询特征信息、重新确定的第二查询特征信息、重新确定的第一查询向量和重新确定的第二查询向量,返回执行确定全景特征图和全景查询向量的步骤,直至满足预设条件;
将最后一次确定的第一查询特征信息作为第一目标查询特征信息,以及将最后一次确定的第二查询特征信息作为第二目标查询特征信息,并基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果。
一种可能的实施方式中,所述第一确定模块,在基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图时,用于:
基于所述第一查询特征信息和所述全景特征信息,确定所述目标图像对应的第一特征图;以及,基于所述第二查询特征信息和所述部件特征信息,确定所述目标图像对应的第二特征图;
将所述第一特征图和所述第二特征图进行拼接,得到所述全景特征图。
一种可能的实施方式中,所述第一确定模块,在基于所述第一查询向量和所述第二查询向量确定全景查询向量时,用于:
将所述第一查询向量和所述第二查询向量进行拼接,得到所述全景查询向量。
一种可能的实施方式中,所述第一确定模块,在基于所述全景特征图对所述全景查询向量进行调整时,用于:
基于所述全景查询向量和所述全景特征图进行动态卷积处理,得到中间查询向量;
基于自注意力机制算法对所述中间查询向量进行处理,得到调整后的全景查询向量。
一种可能的实施方式中,所述第一确定模块,在基于调整后的全景查询向量重新确定第一查询向量和第二查询向量时,用于:
对调整后的全景查询向量进行拆分,得到重新确定的第一查询向量和第二查询向量。
一种可能的实施方式中,所述第一确定模块,在基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果时,用于:
将所述第一目标查询特征信息和所述第二目标查询特征信息进行融合处理,得到所述目标图像的目标分割结果。
一种可能的实施方式中,所述图像分割方法利用图像分割网络完成。
一种可能的实施方式中,所述装置还包括训练模块,所述训练模块用于根据以下步骤训练所述图像分割网络:
获取样本图像以及所述样本图像对应的样本全景分割图像和样本部件分割图像;
将所述样本图像输入至待训练的图像分割网络,确定与所述样本图像对应的第一样本查询特征信息和第二样本查询特征信息;
基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,并基于所述目标损失值对所述待训练的图像分割网络的网络参数进行调整,所述网络参数包括第一查询向量和第二查询向量。
一种可能的实施方式中,所述训练模块,在基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值时,用于:
基于所述第一样本查询特征信息和所述样本全景分割图像,确定全景分割损失;以及,
基于所述第二样本查询特征信息和所述样本部件分割图像,确定部件分割损失;以及,
基于所述第一样本查询特征信息和所述第二样本查询特征信息确定样本全景特征图,并基于样本全景特征图和所述样本图像对应的分类标签,确定分类损失;
基于所述全景分割损失、所述部件分割损失和所述分类损失,确定所述目标损失值。
第四方面,本公开实施例还提供一种车辆控制装置,包括:
第二获取模块,用于获取自动驾驶车辆上部署的图像采集装置采集的待分割图像;
第二确定模块,用于基于第一方面中任一所述的图像分割方法,确定所述待分割图像对应的图像分割结果;
控制模块,用于基于所述图像分割结果对所述自动驾驶车辆进行控制。
第五方面,本公开实施例还提供一种计算机设备,包括:处理器、存储器和总线,所述存储器存储有所述处理器可执行的机器可读指令,当计算机设备运行时,所述处理器与所述存储器之间通过总线通信,所述机器可读指令被所述处理器执行时执行上述第一方面或第二方面中,任一种可能的实施方式中的步骤。
第六方面,本公开实施例还提供一种计算机可读存储介质,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行上述第一方面或第二方面中,任一种可能的实施方式中的步骤。
关于上述图像分割、车辆控制装置、计算机设备、及计算机可读存储介质的效果描述参见上述图像分割、车辆控制方法的说明,这里不再赘述。
本公开实施例提供的图像分割、车辆控制方法、装置、计算机设备及存储介质,通过引入查询向量分别对,目标图像对应的初始特征图进行特征提取后得到的全景特征信息和部件特征信息进行特征查询,并基于特征查询后的第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。这样,通过基于查询向量在一个模型中分别执行全景图像分割任务和部件图像分割任务,相较于使用多个独立建模的模型进行图像分割,提高对所述目标图像的图像分割效果和图像分割效率。
为使本公开的上述目的、特征和优点能更明显易懂,下文特举较佳实施例,并配合所附附图,作详细说明如下。
附图说明
为了更清楚地说明本公开实施例的技术方案,下面将对实施例中所需要使用的附图作简单地介绍,此处的附图被并入说明书中并构成本说明书中的一部分,这些附图示出了符合本公开的实施例,并与说明书一起用于说明本公开的技术方案。应当理解,以下附图仅示出了本公开的某些实施例,因此不应被看作是对范围的限定,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他相关的附图。
图1示出了本公开实施例所提供的一种图像分割方法的流程图;
图2示出了本公开实施例所提供的一种图像分割方法中,确定目标分割结果的具体方法的流程图;
图3示出了本公开实施例所提供的一种图像分割方法的整体流程图;
图4示出了本公开实施例所提供的一种车辆控制方法的流程图;
图5示出了本公开实施例所提供的一种图像分割装置的架构示意图;
图6示出了本公开实施例所提供的一种车辆控制装置的架构示意图;
图7示出了本公开实施例所提供的一种计算机设备的结构示意图。
具体实施方式
为使本公开实施例的目的、技术方案和优点更加清楚,下面将结合本公开实施例中附图,对本公开实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本公开一部分实施例,而不是全部的实施例。通常在此处附图中描述和示出的本公开实施例的组件可以以各种不同的配置来布置和设计。因此,以下对在附图中提供的本公开的实施例的详细描述并非旨在限制要求保护的本公开的范围,而是仅仅表示本公开的选定实施例。基于本公开的实施例,本领域技术人员在没有做出创造性劳动的前提下所获得的所有其他实施例,都属于本公开保护的范围。
应注意到:相似的标号和字母在下面的附图中表示类似项,因此,一旦某一项在一个附图中被定义,则在随后的附图中不需要对其进行进一步定义和解释。
本文中术语“和/或”,仅仅是描述一种关联关系,表示可以存在三种关系,例如,A和/或B,可以表示:单独存在A,同时存在A和B,单独存在B这三种情况。另外,本文中术语“至少一种”表示多种中的任意一种或多种中的至少两种的任意组合,例如,包括A、B、C中的至少一种,可以表示包括从A、B和C构成的集合中选择的任意一个或多个元素。
经研究发现,为了得到更精细化的分割效果,往往会选择使用全景分割模型和部件分割模型分别对待分割图像进行分割,并将全景分割结果和部件分割结果进行融合处理。
具体的,在分别使用全景分割模型和部件分割模型分别对待分割图像进行分割时,所述全景分割模型和所述部件分割模型可以分别使用各自对应的特征提取器进行特征提取,并分别使用各自对应的解码器进行图像分割处理;或者,为了节约计算资源,所述全景分割模型和所述部件分割模型还可以共用同一个特征提取器进行特征提取,并分别使用各自对应的解码器进行图像分割处理,但无论是这两种方式中的哪一种,在进行建模时仍需要建模两个模型,从而导致图像分割效率低下。
基于上述研究,本公开提供了一种图像分割、车辆控制方法、装置、计算机设备及存储介质,通过引入查询向量分别对,目标图像对应的初始特征图进行特征提取后得到的全景特征信息和部件特征信息进行特征查询,并基于特征查询后的第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。这样,通过基于查询向量在一个模型中分别执行全景图像分割任务和部件图像分割任务,相较于使用多个独立建模的模型进行图像分割,提高对所述目标图像的图像分割效果和图像分割效率。
为便于对本实施例进行理解,首先对本公开实施例所公开的一种图像分割方法进行详细介绍,本公开实施例所提供的图像分割方法的执行主体一般为具有一定计算能力的计算机设备,该计算机设备例如包括:终端设备或服务器或其它处理设备,终端设备可以为用户设备(User Equipment,UE)、移动设备、用户终端、终端、蜂窝电话、无绳电话、个人数字助理(Personal Digital Assistant,PDA)、手持设备、计算设备、车载设备、可穿戴设备等。在一些可能的实现方式中,该图像分割方法可以通过处理器调用存储器中存储的计算机可读指令的方式来实现。
参见图1所示,为本公开实施例提供的图像分割方法的流程图,所述方法包括S101~S104,其中:
S101:获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图。
S102:分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息。
S103:基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息。
S104:基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
以下是对上述步骤的详细介绍。
针对S101,所述目标图像例如可以是自动驾驶车辆上部署的图像采集装置采集的待分割图像。
具体的,在对所述目标图像进行特征提取时,可以使用骨干网络backbone对所述目标图像进行特征提取,所述骨干网络的网络结构可以由卷积神经网络(ConvolutionalNeural Networks,CNN)和特征金字塔网络(Feature Pyramid Networks,FPN)组成,通过所述骨干网络对所述目标图像进行特征提取,可以得到的所述目标图像中各像素点对应的多层级特征表达(即所述初始特征图)。
S102:分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息。
这里,所述全景特征信息可以包括前景特征信息和背景特征信息,其中,所述前景特征信息表征图像中前景物体的特征,比如汽车的特征;所述背景特征信息表征图像中背景物体的特征,比如马路的特征;所述部件特征信息表征图像中可数物体的部件的特征,比如汽车车灯的特征。
具体的,在对所述初始特征图进行全景特征提取时,可以使用第一解码器对所述初始特征图进行全景特征提取,从而得到所述目标图像对应的全景特征信息;在对所述初始特征图进行部件特征提取时,可以使用第二解码器对所述初始特征图进行部件特征提取,从而可以得到所述目标图像对应的部件特征信息。
其中,所述第一解码器和第二解码器可以为Transform模型中的解码器(decoder),所述第一解码器和第二解码器的网络结构可以由FPN组成,通过将多层级特征融合成一个尺度的特征,即可得到对应的特征信息。
进一步的,为了得到更为准确的部件特征信息,可以在所述第二解码器中加入对齐处理机制,以使得特征在解码过程中的上采样阶段更为对齐,从而可以保留目标图像中的更多细节信息。
S103:基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息。
这里,所述第一查询向量可以包括前景查询向量和背景查询向量;所述第一查询特征信息可以包括前景查询信息和背景查询信息,所述第一查询特征信息和第二查询特征信息通过掩模图像(Mask)进行表征,所述掩模图像可以用于表示图像分割结果。
具体的,所述第一查询向量和所述第二查询向量为预先设置的可学习的参数(也即在图像分割的处理过程中可以被调整的参数),所述第一查询向量和所述第二查询向量分别对应的初始值可以是由开发人员设置的,以本公开实施例中的图像分割方法利用图像分割网络完成为例,所述第一查询向量和所述第二查询向量分别对应的初始值可以是通过训练所述图像分割网络得到,训练完成之后即可使用训练好的图像分割网络以及对应的第一查询向量和所述第二查询向量对待分割的目标图像进行图像分割。
一种可能的实施方式中,在进行特征查询时,可以基于所述前景查询向量对所述全景特征信息进行卷积处理,得到所述前景查询信息;以及,基于所述背景查询向量对所述全景特征信息进行卷积处理,得到所述背景查询信息。
这样,基于前景查询向量对全景特征信息进行卷积处理,可以从所述全景特征信息中提取出与所述目标图像中的前景相关的深层特征(也即前景查询信息);基于背景查询向量对全景特征信息进行卷积处理,可以从所述全景特征信息中提取出与所述目标图像中的背景相关的深层特征(也即背景查询信息)。
S104:基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
一种可能的实施方式中,如图2所示,在确定目标分割结果时,可以采用下述步骤:
S201:基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图;以及,基于所述第一查询向量和所述第二查询向量确定全景查询向量。
一种可能的实施方式中,在确定所述目标图像对应的全景特征图时,可以基于所述第一查询特征信息和所述全景特征信息,确定所述目标图像对应的第一特征图;以及,基于所述第二查询特征信息和所述部件特征信息,确定所述目标图像对应的第二特征图;将所述第一特征图和所述第二特征图进行拼接,得到所述全景特征图。
具体的,由于第一查询特征信息和第二查询特征信息的尺寸不同,后续难以进行拼接,因此在确定所述第一特征图时,可以基于所述第一查询特征信息和所述全景特征信息进行池化处理,得到所述目标图像对应的第一特征图;在确定所述第二特征图时,可以基于所述第二查询特征信息和所述部件特征信息进行池化处理,得到所述目标图像对应的第二特征图,以确保所述第一特征图和第二特征图的尺寸相同,从而能够进行后续的特征图拼接,其中,所述池化处理可以是内积处理等池化处理方式。
示例性的,以所述池化处理的方式为内积处理为例,通过确定所述第第一查询特征信息和所述全景特征信息的内积,即可得到所述目标图像对应的第一特征图;通过确定所述第二查询特征信息和所述部件特征信息的内积,即可得到所述目标图像对应的第二特征图。
这样,通过确定第一特征图和第二特征图,并对所述第一特征图和第二特征图进行拼接得到全景特征图,使得后续对查询向量进行调整时只需要处理一个特征图即可,从而可以提高图像分割效率。
一种可能的实施方式中,在确定所述全景查询向量时,可以将所述第一查询向量和所述第二查询向量进行拼接,得到所述全景查询向量。
这样,通过将第一查询向量和第二查询向量进行拼接得到全景查询向量,使得后续对全景查询向量进行调整时只需要处理一个查询向量即可,从而可以提高图像分割效率。
S202:基于所述全景特征图对所述全景查询向量进行调整,并基于调整后的全景查询向量重新确定第一查询向量和第二查询向量。
一种可能的实施方式中,在对所述全景查询向量进行调整时,可以采用下述步骤A1~A2:
A1:基于所述全景查询向量和所述全景特征图进行动态卷积处理,得到中间查询向量。
这里,在进行动态卷积处理时,可以基于门控机制算法对所述全景查询向量和所述全景特征图进行加权求和,得到所述中间查询向量。
具体的,可以基于两个不同的门函数(gate function),分别确定所述全景查询向量和所述全景特征图分别对应的权重,并基于确定的权重对所述全景查询向量和所述全景特征图进行加权求和,得到所述中间查询向量。
A2:基于自注意力机制算法对所述中间查询向量进行处理,得到调整后的全景查询向量。
这里,所述自注意力机制算法例如可以是多头自注意力机制算法(Multi-headedSelf-attention,MHSA)等。
进一步的,在基于调整后的全景查询向量重新确定第一查询向量和第二查询向量时,可以对调整后的全景查询向量进行拆分,得到重新确定的第一查询向量和第二查询向量。
具体的,在对调整后的全景查询向量进行拆分时,可以按照所述第一查询向量和所述第二查询向量拼接时的拼接顺序(所述拼接顺序可以为预设的拼接顺序),对所述全景查询向量进行拆分,以重新确定第一查询向量和第二查询向量。
这样,基于动态卷积处理和自注意力机制算法对所述全景查询向量进行调整,并对调整后的全景查询向量进行拆分,可以完成对所述第一查询向量和所述第二查询向量的一次更新,并可以通过后续的更新,使得查询向量能够更好的提取出目标图像对应的全景特征(第一查询特征信息)和部件特征(第二查询特征信息),从而提高图像分割效果。
S203:基于重新确定的第一查询向量重新确定第一查询特征信息,以及基于重新确定的第二查询向量,重新确定第二查询特征信息。
这里,重新确定第一查询特征信息和第二查询特征信息时的具体步骤,可以参照上文S103的相关内容,在此不在赘述。
S204:判断是否满足预设条件。
这里,所述预设条件可以是返回执行确定全景特征图和全景查询向量的次数达到预设次数,所述预设次数例如可以是3次。
示例性的,以所述预设次数为3为例,可以将第3次确定的第一查询特征信息作为所述第一目标查询特征信息,将第3次确定的第一查询特征信息作为所述第二目标查询特征信息,并基于所述第一目标查询特征信息和所述第二目标查询特征信息确定所述目标分割结果。
具体的,可以基于重新确定的第一查询特征信息、重新确定的第二查询特征信息、重新确定的第一查询向量和重新确定的第二查询向量,确定是否满足预设条件,并在不满足预设条件的情况下,返回执行S201,直至满足预设条件后执行S205,也即基于重新确定的第一查询特征信息和重新确定的第二查询特征信息,返回执行确定全景特征图和全景查询向量的步骤,直至满足预设条件。
S205:将最后一次确定的第一查询特征信息作为第一目标查询特征信息,以及将最后一次确定的第二查询特征信息作为第二目标查询特征信息,并基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果。
进一步的,在确定所述目标图像的目标分割结果时,可以将所述第一目标查询特征信息和所述第二目标查询特征信息进行融合处理,得到所述目标图像的目标分割结果。
具体的,在进行融合处理时,可以对所述第一目标特征信息中包含的前景查询信息和背景查询信息进行融合处理,得到所述目标图像对应的全景分割结果;根据所述第二目标查询特征信息和所述全景分割结果进行融合处理,得到融合了全景分割结果和部件分割结果的目标分割结果。
这样,通过对第一目标查询特征信息和第二目标查询特征信息进行融合处理,可以得到融合了全景分割结果和部件分割结果的目标分割结果。
一种可能的实施方式中,本公开实施例提供的所述图像分割方法可以利用图像分割网络完成,在对所述图像分割网络进行训练时,可以采用下述步骤B1~B3:
B1:获取样本图像以及所述样本图像对应的样本全景分割图像和样本部件分割图像。
这里,可以获取与所述样本图像对应的第一分割图像和第二分割图像作为训练时的监督数据,其中所述第一分割图像为样本全景分割图像,所述第二分割图像为样本部件分割图像。
B2:将所述样本图像输入至待训练的图像分割网络,确定与所述样本图像对应的第一样本查询特征信息和第二样本查询特征信息。
B3:基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,并基于所述目标损失值对所述待训练的图像分割网络的网络参数进行调整,所述网络参数包括第一查询向量和第二查询向量。
这里,由于在所述图像分割网络中所述第一查询特征信息和所述第二查询特征信息可以被多次重新确定,为了提高训练效果,可以将每次确定的第一查询向量和第二查询向量均用于确定本次训练的目标损失值。
一种可能的实施方式中,在确定所述目标损失值,可以采用下述步骤D1~D2:
D1:基于所述第一样本查询特征信息和所述样本全景分割图像,确定全景分割损失;以及,基于所述第二样本查询特征信息和所述样本部件分割图像,确定部件分割损失;以及,基于所述第一样本查询特征信息和所述第二样本查询特征信息确定样本全景特征图,并基于样本全景特征图和所述样本图像对应的分类标签,确定分类损失。
D2:基于所述全景分割损失、所述部件分割损失和所述分类损失,确定所述目标损失值。
具体的,在基于所述全景分割损失、所述部件分割损失和所述分类损失,确定所述目标损失值时,可以按照预设的权重对所述全景分割损失、所述部件分割损失和所述分类损失进行加权求和。
这样,基于分割损失和分类损失确定目标损失值,可以使得后续在基于所述目标损失值对待训练的图像分割网络的网络参数进行调整后,图像分割网络的网络精度更高,从而提高图像分割效率。
下面,将结合具体的实施方式,对上述图像分割方法进行整体描述。参见图3所示,为本公开实施例提供的一种图像分割方法的整体流程图。
1、基于骨干网络backbone对待分割的目标图像进行特征提取,得到所述目标图像对应的初始特征图。
2、将所述初始特征图输入至第一解码器“Decoder”进行全景特征提取,得到所述目标图像对应的全景特征信息Fs;以及,将所述初始特征图输入至第二解码器“AlignedDecoder”进行部件特征提取,得到所述目标图像对应的部件特征信息Fp。
其中,所述第二解码器为加入对齐处理(Align)机制的解码器,以使得特征在解码过程中的上采样阶段更为对齐,从而可以保留目标图像中的更多细节信息,图2中虚线框内表示第一解码器和第二解码器能够对所述初始特征图进行解耦处理(Decoupled),从而得到所述全景特征信息Fs和所述部件特征信息Fp。
3、基于与所述全景特征信息Fs对应的第一查询向量
Figure BDA0003557405180000151
对所述全景特征信息Fs进行特征查询,得到用于表征全景分割结果的第一查询特征信息
Figure BDA0003557405180000152
以及,基于与所述部件特征信息Fp对应的第二查询向量
Figure BDA0003557405180000153
对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息
Figure BDA0003557405180000154
其中,i表示迭代次数,第一次执行时i=1,一次图像分割过程中执行的迭代次数为预设次数。
4、基于所述第一查询特征信息
Figure BDA0003557405180000155
和所述全景特征信息Fs,确定所述目标图像对应的第一特征图
Figure BDA0003557405180000156
以及,基于所述第二查询特征信息
Figure BDA0003557405180000157
和所述部件特征信息Fp,确定所述目标图像对应的第二特征图
Figure BDA0003557405180000158
这里,在确定所述第一特征图和所述第二特征图时,可以使用如下公式1:
Figure BDA0003557405180000159
其中,H和W分别表示特征查询信息对应的特征图的宽和高,u和v表示对应的特征空间图的位置坐标。
5、将所述第一特征图
Figure BDA00035574051800001510
和所述第二特征图
Figure BDA00035574051800001511
进行拼接,得到全景特征图
Figure BDA00035574051800001512
以及,将所述第一查询向量
Figure BDA00035574051800001513
和所述第二查询向量
Figure BDA00035574051800001514
进行拼接,得到全景查询向量
Figure BDA00035574051800001515
6、基于所述全景查询向量
Figure BDA00035574051800001516
和所述全景特征图
Figure BDA00035574051800001517
进行动态卷积处理,得到中间查询向量
Figure BDA00035574051800001518
并基于自注意力机制算法对所述中间查询向量
Figure BDA00035574051800001519
进行处理,得到调整后的全景查询向量
Figure BDA00035574051800001520
这里,在确定中间查询向量时,可以使用如下公式2:
Figure BDA00035574051800001521
其中,DynamicConv表示对所述全景查询向量
Figure BDA0003557405180000161
和所述全景特征图
Figure BDA0003557405180000162
进行动态卷积处理。
示例性的,在进行动态卷积处理时,可以使用如下公式3:
Figure BDA0003557405180000163
其中,Gatex和Gateq为两个不同的门函数(gate function),分别确定所述全景查询向量
Figure BDA0003557405180000164
和所述全景特征图
Figure BDA0003557405180000165
分别对应的权重,并基于确定的权重对所述全景查询向量
Figure BDA0003557405180000166
和所述全景特征图
Figure BDA0003557405180000167
进行加权求和,得到所述中间查询向量
Figure BDA0003557405180000168
这里,在基于中间查询向量确定更新后的全景查询向量
Figure BDA0003557405180000169
时,可以使用如下公式4:
Figure BDA00035574051800001610
其中,FFN表示前馈神经网络处理(FeedForward Neural Network),MHSA表示多头自注意力机制算法。
7、基于更新后的全景查询向量
Figure BDA00035574051800001616
返回执行确定第一查询特征信息和第二查询特征信息的步骤,直至满足预设条件;并基于最后一次确定的第一目标查询特征信息和第二目标查询特征信息,确定所述目标图像的目标分割结果。
这里,所述预设条件为i=3,即最后一次确定的第一目标查询特征信息为
Figure BDA00035574051800001611
第二目标查询特征信息为
Figure BDA00035574051800001612
最终确定的目标分割结果为将所述第一目标查询特征信息
Figure BDA00035574051800001613
和所述第二目标查询特征信息
Figure BDA00035574051800001614
进行融合处理后得到的
Figure BDA00035574051800001615
其中,3表示迭代次数为3次。
参见图4所示,为本公开实施例提供的车辆控制方法的流程图,所述方法包括S401~S403,其中:
S401:获取自动驾驶车辆上部署的图像采集装置采集的待分割图像。
S402:基于本公开实施例任一所述的图像分割方法,确定所述待分割图像对应的图像分割结果。
S403:基于所述图像分割结果对所述自动驾驶车辆进行控制。
这里,对所述自动驾驶车辆进行控制,例如包括控制车辆加速、减速、转向、制动等,或者可以播放语音提示信息,以提示驾驶员控制车辆加速、减速、转向、制动等。
本公开实施例提供的图像分割、车辆控制方法,通过引入查询向量分别对,目标图像对应的初始特征图进行特征提取后得到的全景特征信息和部件特征信息进行特征查询,并基于特征查询后的第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。这样,通过基于查询向量在一个模型中分别执行全景图像分割任务和部件图像分割任务,相较于使用多个独立建模的模型进行图像分割,提高对所述目标图像的图像分割效果和图像分割效率。
本领域技术人员可以理解,在具体实施方式的上述方法中,各步骤的撰写顺序并不意味着严格的执行顺序而对实施过程构成任何限定,各步骤的具体执行顺序应当以其功能和可能的内在逻辑确定。
基于同一发明构思,本公开实施例中还提供了与图像分割方法对应的图像分割装置,由于本公开实施例中的装置解决问题的原理与本公开实施例上述图像分割方法相似,因此装置的实施可以参见方法的实施,重复之处不再赘述。
参照图5所示,为本公开实施例提供的一种图像分割装置的架构示意图,所述装置包括:第一获取模块501、提取模块502、查询模块503、第一确定模块504;其中,
第一获取模块501,用于获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;
提取模块502,用于分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;
查询模块503,用于基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;
第一确定模块504,用于基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
一种可能的实施方式中,所述第一确定模块504,在基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果时,用于:
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图;以及,基于所述第一查询向量和所述第二查询向量确定全景查询向量;
基于所述全景特征图对所述全景查询向量进行调整,并基于调整后的全景查询向量重新确定第一查询向量和第二查询向量;
基于重新确定的第一查询向量重新确定第一查询特征信息,以及基于重新确定的第二查询向量,重新确定第二查询特征信息;
基于重新确定的第一查询特征信息、重新确定的第二查询特征信息、重新确定的第一查询向量和重新确定的第二查询向量,返回执行确定全景特征图和全景查询向量的步骤,直至满足预设条件;
将最后一次确定的第一查询特征信息作为第一目标查询特征信息,以及将最后一次确定的第二查询特征信息作为第二目标查询特征信息,并基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果。
一种可能的实施方式中,所述第一确定模块504,在基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图时,用于:
基于所述第一查询特征信息和所述全景特征信息,确定所述目标图像对应的第一特征图;以及,基于所述第二查询特征信息和所述部件特征信息,确定所述目标图像对应的第二特征图;
将所述第一特征图和所述第二特征图进行拼接,得到所述全景特征图。
一种可能的实施方式中,所述第一确定模块504,在基于所述第一查询向量和所述第二查询向量确定全景查询向量时,用于:
将所述第一查询向量和所述第二查询向量进行拼接,得到所述全景查询向量。
一种可能的实施方式中,所述第一确定模块504,在基于所述全景特征图对所述全景查询向量进行调整时,用于:
基于所述全景查询向量和所述全景特征图进行动态卷积处理,得到中间查询向量;
基于自注意力机制算法对所述中间查询向量进行处理,得到调整后的全景查询向量。
一种可能的实施方式中,所述第一确定模块504,在基于调整后的全景查询向量重新确定第一查询向量和第二查询向量时,用于:
对调整后的全景查询向量进行拆分,得到重新确定的第一查询向量和第二查询向量。
一种可能的实施方式中,所述第一确定模块504,在基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果时,用于:
将所述第一目标查询特征信息和所述第二目标查询特征信息进行融合处理,得到所述目标图像的目标分割结果。
一种可能的实施方式中,所述图像分割方法利用图像分割网络完成。
一种可能的实施方式中,所述装置还包括训练模块505,所述训练模块505用于根据以下步骤训练所述图像分割网络:
获取样本图像以及所述样本图像对应的样本全景分割图像和样本部件分割图像;
将所述样本图像输入至待训练的图像分割网络,确定与所述样本图像对应的第一样本查询特征信息和第二样本查询特征信息;
基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,并基于所述目标损失值对所述待训练的图像分割网络的网络参数进行调整,所述网络参数包括第一查询向量和第二查询向量。
一种可能的实施方式中,所述训练模块505,在基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值时,用于:
基于所述第一样本查询特征信息和所述样本全景分割图像,确定全景分割损失;以及,
基于所述第二样本查询特征信息和所述样本部件分割图像,确定部件分割损失;以及,
基于所述第一样本查询特征信息和所述第二样本查询特征信息确定样本全景特征图,并基于样本全景特征图和所述样本图像对应的分类标签,确定分类损失;
基于所述全景分割损失、所述部件分割损失和所述分类损失,确定所述目标损失值。
本公开实施例提供的图像分割装置,通过引入查询向量分别对,目标图像对应的初始特征图进行特征提取后得到的全景特征信息和部件特征信息进行特征查询,并基于特征查询后的第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。这样,通过基于查询向量在一个模型中分别执行全景图像分割任务和部件图像分割任务,相较于使用多个独立建模的模型进行图像分割,提高对所述目标图像的图像分割效果和图像分割效率。
参照图6所示,为本公开实施例提供的一种车辆控制装置的架构示意图,所述装置包括:第二获取模块601、第二确定模块602、控制模块603;其中,
第二获取模块601,用于获取自动驾驶车辆上部署的图像采集装置采集的待分割图像;
第二确定模块602,用于基于本公开任一实施例所述的图像分割方法,确定所述待分割图像对应的图像分割结果;
控制模块603,用于基于所述图像分割结果对所述自动驾驶车辆进行控制。
关于装置中的各模块的处理流程、以及各模块之间的交互流程的描述可以参照上述方法实施例中的相关说明,这里不再详述。
基于同一技术构思,本公开实施例还提供了一种计算机设备。参照图7所示,为本公开实施例提供的计算机设备700的结构示意图,包括处理器701、存储器702、和总线703。其中,存储器702用于存储执行指令,包括内存7021和外部存储器7022;这里的内存7021也称内存储器,用于暂时存放处理器701中的运算数据,以及与硬盘等外部存储器7022交换的数据,处理器701通过内存7021与外部存储器7022进行数据交换,当计算机设备700运行时,处理器701与存储器702之间通过总线703通信,使得处理器701在执行以下指令:
获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;
分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;
基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果;或者,
使得处理器701在执行以下指令:
获取自动驾驶车辆上部署的图像采集装置采集的待分割图像;
基于本公开任一实施例所述的图像分割方法,确定所述待分割图像对应的图像分割结果;
基于所述图像分割结果对所述自动驾驶车辆进行控制。
本公开实施例还提供一种计算机可读存储介质,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行上述方法实施例中所述的图像分割、车辆控制方法的步骤。其中,该存储介质可以是易失性或非易失的计算机可读取存储介质。
本公开实施例还提供一种计算机程序产品,该计算机程序产品承载有程序代码,所述程序代码包括的指令可用于执行上述方法实施例中所述的图像分割、车辆控制方法的步骤,具体可参见上述方法实施例,在此不再赘述。
其中,上述计算机程序产品可以具体通过硬件、软件或其结合的方式实现。在一个可选实施例中,所述计算机程序产品具体体现为计算机存储介质,在另一个可选实施例中,计算机程序产品具体体现为软件产品,例如软件开发包(Software Development Kit,SDK)等等。
所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的系统和装置的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。在本公开所提供的几个实施例中,应该理解到,所揭露的系统、装置和方法,可以通过其它的方式实现。以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,又例如,多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些通信接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
另外,在本公开各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。
所述功能如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个处理器可执行的非易失的计算机可读取存储介质中。基于这样的理解,本公开的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本公开各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(Read-OnlyMemory,ROM)、随机存取存储器(Random Access Memory,RAM)、磁碟或者光盘等各种可以存储程序代码的介质。
最后应说明的是:以上所述实施例,仅为本公开的具体实施方式,用以说明本公开的技术方案,而非对其限制,本公开的保护范围并不局限于此,尽管参照前述实施例对本公开进行了详细的说明,本领域的普通技术人员应当理解:任何熟悉本技术领域的技术人员在本公开揭露的技术范围内,其依然可以对前述实施例所记载的技术方案进行修改或可轻易想到变化,或者对其中部分技术特征进行等同替换;而这些修改、变化或者替换,并不使相应技术方案的本质脱离本公开实施例技术方案的精神和范围,都应涵盖在本公开的保护范围之内。因此,本公开的保护范围应所述以权利要求的保护范围为准。

Claims (15)

1.一种图像分割方法,其特征在于,包括:
获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;
分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;
基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
2.根据权利要求1所述的方法,其特征在于,所述基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果,包括:
基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图;以及,基于所述第一查询向量和所述第二查询向量确定全景查询向量;
基于所述全景特征图对所述全景查询向量进行调整,并基于调整后的全景查询向量重新确定第一查询向量和第二查询向量;
基于重新确定的第一查询向量重新确定第一查询特征信息,以及基于重新确定的第二查询向量,重新确定第二查询特征信息;
基于重新确定的第一查询特征信息、重新确定的第二查询特征信息、重新确定的第一查询向量和重新确定的第二查询向量,返回执行确定全景特征图和全景查询向量的步骤,直至满足预设条件;
将最后一次确定的第一查询特征信息作为第一目标查询特征信息,以及将最后一次确定的第二查询特征信息作为第二目标查询特征信息,并基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果。
3.根据权利要求2所述的方法,其特征在于,所述基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像对应的全景特征图,包括:
基于所述第一查询特征信息和所述全景特征信息,确定所述目标图像对应的第一特征图;以及,基于所述第二查询特征信息和所述部件特征信息,确定所述目标图像对应的第二特征图;
将所述第一特征图和所述第二特征图进行拼接,得到所述全景特征图。
4.根据权利要求2或3所述的方法,其特征在于,所述基于所述第一查询向量和所述第二查询向量确定全景查询向量,包括:
将所述第一查询向量和所述第二查询向量进行拼接,得到所述全景查询向量。
5.根据权利要求2~4任一所述的方法,其特征在于,所述基于所述全景特征图对所述全景查询向量进行调整,包括:
基于所述全景查询向量和所述全景特征图进行动态卷积处理,得到中间查询向量;
基于自注意力机制算法对所述中间查询向量进行处理,得到调整后的全景查询向量。
6.根据权利要求2~5任一所述的方法,其特征在于,所述基于调整后的全景查询向量重新确定第一查询向量和第二查询向量,包括:
对调整后的全景查询向量进行拆分,得到重新确定的第一查询向量和第二查询向量。
7.根据权利要求2~6任一所述的方法,其特征在于,所述基于所述第一目标查询特征信息和所述第二目标查询特征信息,确定所述目标图像的目标分割结果,包括:
将所述第一目标查询特征信息和所述第二目标查询特征信息进行融合处理,得到所述目标图像的目标分割结果。
8.根据权利要求1~7任一所述的方法,其特征在于,所述图像分割方法利用图像分割网络完成。
9.根据权利要求8所述的方法,其特征在于,所述图像分割网络采用以下步骤训练得到:
获取样本图像以及所述样本图像对应的样本全景分割图像和样本部件分割图像;
将所述样本图像输入至待训练的图像分割网络,确定与所述样本图像对应的第一样本查询特征信息和第二样本查询特征信息;
基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,并基于所述目标损失值对所述待训练的图像分割网络的网络参数进行调整,所述网络参数包括第一查询向量和第二查询向量。
10.根据权利要求9所述的方法,其特征在于,所述基于所述第一样本查询特征信息、所述第二样本查询特征信息、所述样本全景分割图像和所述样本部件分割图像,确定本次训练的目标损失值,包括:
基于所述第一样本查询特征信息和所述样本全景分割图像,确定全景分割损失;以及,
基于所述第二样本查询特征信息和所述样本部件分割图像,确定部件分割损失;以及,
基于所述第一样本查询特征信息和所述第二样本查询特征信息确定样本全景特征图,并基于样本全景特征图和所述样本图像对应的分类标签,确定分类损失;
基于所述全景分割损失、所述部件分割损失和所述分类损失,确定所述目标损失值。
11.一种车辆控制方法,其特征在于,包括:
获取自动驾驶车辆上部署的图像采集装置采集的待分割图像;
基于权利要求1~10任一所述的图像分割方法,确定所述待分割图像对应的图像分割结果;
基于所述图像分割结果对所述自动驾驶车辆进行控制。
12.一种图像分割装置,其特征在于,包括:
第一获取模块,用于获取待分割的目标图像,并对所述目标图像进行特征提取,得到所述目标图像对应的初始特征图;
提取模块,用于分别对所述初始特征图进行全景特征提取和部件特征提取,得到所述目标图像对应的全景特征信息和部件特征信息;
查询模块,用于基于与所述全景特征信息对应的第一查询向量对所述全景特征信息进行特征查询,得到用于表征全景分割结果的第一查询特征信息;以及,基于与所述部件特征信息对应的第二查询向量对所述部件特征信息进行特征查询,得到用于表征部件分割结果的第二查询特征信息;
第一确定模块,用于基于所述第一查询特征信息和所述第二查询特征信息,确定所述目标图像的目标分割结果。
13.一种车辆控制装置,其特征在于,包括:
第二获取模块,用于获取自动驾驶车辆上部署的图像采集装置采集的待分割图像;
第二确定模块,用于基于权利要求1~10任一所述的图像分割方法,确定所述待分割图像对应的图像分割结果;
控制模块,用于基于所述图像分割结果对所述自动驾驶车辆进行控制。
14.一种计算机设备,其特征在于,包括:处理器、存储器和总线,所述存储器存储有所述处理器可执行的机器可读指令,当计算机设备运行时,所述处理器与所述存储器之间通过总线通信,所述机器可读指令被所述处理器执行时执行如权利要求1至10任一所述的图像分割方法的步骤;或者,执行如权利要求11所述的车辆控制方法的步骤。
15.一种计算机可读存储介质,其特征在于,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行如权利要求1至10任一所述的图像分割方法的步骤;或者,执行如权利要求11所述的车辆控制方法的步骤。
CN202210283973.5A 2022-03-21 2022-03-21 图像分割方法、车辆控制方法及装置 Pending CN114612667A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202210283973.5A CN114612667A (zh) 2022-03-21 2022-03-21 图像分割方法、车辆控制方法及装置

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202210283973.5A CN114612667A (zh) 2022-03-21 2022-03-21 图像分割方法、车辆控制方法及装置

Publications (1)

Publication Number Publication Date
CN114612667A true CN114612667A (zh) 2022-06-10

Family

ID=81864554

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202210283973.5A Pending CN114612667A (zh) 2022-03-21 2022-03-21 图像分割方法、车辆控制方法及装置

Country Status (1)

Country Link
CN (1) CN114612667A (zh)

Cited By (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116778170A (zh) * 2023-08-25 2023-09-19 安徽蔚来智驾科技有限公司 点云全景分割方法、控制装置、可读存储介质及车辆

Cited By (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116778170A (zh) * 2023-08-25 2023-09-19 安徽蔚来智驾科技有限公司 点云全景分割方法、控制装置、可读存储介质及车辆
CN116778170B (zh) * 2023-08-25 2023-11-07 安徽蔚来智驾科技有限公司 点云全景分割方法、控制装置、可读存储介质及车辆

Similar Documents

Publication Publication Date Title
KR102529574B1 (ko) 소프트 교차-엔트로피 손실을 갖는 시맨틱 분할
CN111858869B (zh) 一种数据匹配方法、装置、电子设备和存储介质
CN111597884A (zh) 面部动作单元识别方法、装置、电子设备及存储介质
CN113344998B (zh) 深度检测方法、装置、计算机设备及存储介质
CN109034086B (zh) 车辆重识别方法、装置及系统
CN110245621B (zh) 人脸识别装置及图像处理方法、特征提取模型、存储介质
CN110378305B (zh) 茶叶病害识别方法、设备、存储介质及装置
CN111108508B (zh) 脸部情感识别方法、智能装置和计算机可读存储介质
CN112862828A (zh) 一种语义分割方法、模型训练方法及装置
CN112560622B (zh) 虚拟对象动作控制方法、装置及电子设备
CN111507226A (zh) 道路图像识别模型建模方法、图像识别方法及电子设备
CN112017162B (zh) 病理图像处理方法、装置、存储介质和处理器
CN114238904A (zh) 身份识别方法、双通道超分模型的训练方法及装置
CN112818955A (zh) 一种图像分割方法、装置、计算机设备以及存储介质
CN114266894A (zh) 一种图像分割方法、装置、电子设备及存储介质
CN106384071A (zh) 一种二维码扫描识别处理方法及装置
CN116994021A (zh) 图像检测方法、装置、计算机可读介质及电子设备
CN114612667A (zh) 图像分割方法、车辆控制方法及装置
CN113221977A (zh) 一种基于抗混叠语义重构的小样本语义分割方法
CN116468947A (zh) 刀具图像识别方法、装置、计算机设备及存储介质
CN115830618A (zh) 一种文本识别方法、装置、计算机设备及存储介质
CN114973424A (zh) 特征提取模型训练、手部动作识别方法、装置及电子设备
CN112419249B (zh) 一种特殊服饰图片转化方法、终端设备及存储介质
CN112288748B (zh) 一种语义分割网络训练、图像语义分割方法及装置
CN117612231B (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