CN114565760A - 图像分割方法、模型的训练方法、装置、电子设备及介质 - Google Patents
图像分割方法、模型的训练方法、装置、电子设备及介质 Download PDFInfo
- Publication number
- CN114565760A CN114565760A CN202210174233.8A CN202210174233A CN114565760A CN 114565760 A CN114565760 A CN 114565760A CN 202210174233 A CN202210174233 A CN 202210174233A CN 114565760 A CN114565760 A CN 114565760A
- Authority
- CN
- China
- Prior art keywords
- information
- scale
- information corresponding
- image
- correlation
- 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.)
- Granted
Links
- 238000003709 image segmentation Methods 0.000 title claims abstract description 166
- 238000000034 method Methods 0.000 title claims abstract description 114
- 238000012549 training Methods 0.000 title claims abstract description 35
- 238000004590 computer program Methods 0.000 claims description 13
- 238000013473 artificial intelligence Methods 0.000 abstract description 4
- 238000013135 deep learning Methods 0.000 abstract description 2
- 230000000875 corresponding effect Effects 0.000 description 477
- 239000013598 vector Substances 0.000 description 190
- 238000012545 processing Methods 0.000 description 40
- 238000000605 extraction Methods 0.000 description 33
- 230000008569 process Effects 0.000 description 31
- 238000013528 artificial neural network Methods 0.000 description 27
- 230000004927 fusion Effects 0.000 description 14
- 238000010586 diagram Methods 0.000 description 13
- 238000004891 communication Methods 0.000 description 11
- 230000002596 correlated effect Effects 0.000 description 6
- 230000011218 segmentation Effects 0.000 description 6
- 230000006870 function Effects 0.000 description 5
- 238000013527 convolutional neural network Methods 0.000 description 4
- 230000000007 visual effect Effects 0.000 description 4
- 230000008859 change Effects 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 230000004048 modification Effects 0.000 description 3
- 238000012986 modification Methods 0.000 description 3
- 230000003287 optical effect Effects 0.000 description 3
- 230000006872 improvement Effects 0.000 description 2
- 230000003993 interaction Effects 0.000 description 2
- 238000010606 normalization Methods 0.000 description 2
- 238000012546 transfer Methods 0.000 description 2
- 238000003491 array Methods 0.000 description 1
- 230000004888 barrier function Effects 0.000 description 1
- 230000005540 biological transmission Effects 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 230000001413 cellular effect Effects 0.000 description 1
- 238000012512 characterization method Methods 0.000 description 1
- 238000010276 construction Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 239000004973 liquid crystal related substance Substances 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 238000004519 manufacturing process Methods 0.000 description 1
- 239000000463 material Substances 0.000 description 1
- 239000011159 matrix material Substances 0.000 description 1
- 230000007246 mechanism Effects 0.000 description 1
- 239000013307 optical fiber Substances 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 239000004065 semiconductor Substances 0.000 description 1
- 230000001953 sensory effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 238000004804 winding Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Computing Systems (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Life Sciences & Earth Sciences (AREA)
- Molecular Biology (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Image Analysis (AREA)
- Information Retrieval, Db Structures And Fs Structures Therefor (AREA)
Abstract
本公开提供了一种图像分割方法、模型的训练方法、装置、电子设备及介质,涉及人工智能技术领域,尤其涉及计算机视觉、图像识别和深度学习技术领域。具体实现方案为:根据查询图像的查询图像数据、支持图像的支持图像数据和与支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息,特征信息包括查询特征信息、支持特征信息和掩码特征信息;根据至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,相关信息包括以下至少一项:查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息;根据至少一个第二预定尺度的特征信息和至少一个第一预定尺度的相关信息,得到查询图像的图像分割结果。
Description
技术领域
本公开涉及人工智能技术领域,尤其涉及计算机视觉、图像识别和深度学习技术领域。具体地,涉及一种图像分割方法、模型的训练方法、装置、电子设备及介质。
背景技术
图像分割是计算机视觉领域的研究方向之一。图像分割是指将图像划分成各自满足相似性准则或具有某种同质特征的连通区域的过程。图像分割可以包括小样本分割(即少样本分割)(Few-Shot Segmentation,FSS)。
随着人工智能技术的发展,小样本分割在多个领域得到发展。例如,自动驾驶和医学诊断等。
发明内容
本公开提供了一种图像分割方法、模型的训练方法、装置、电子设备及介质。
根据本公开的一方面,提供了一种图像分割方法,包括:根据查询图像的查询图像数据、支持图像的支持图像数据和与上述支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息,其中,上述特征信息包括上述查询图像的查询特征信息、上述支持图像的支持特征信息和上述掩码图像的掩码特征信息;根据上述至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,其中,上述相关信息包括以下至少一项:上述查询图像和上述支持图像之间的互相关语义信息以及上述查询图像和上述支持图像之间的互关联信息,上述互相关语义信息是根据上述支持图像的第一自相关语义信息和上述查询图像的第二自相关语义信息确定的,上述互关联信息是根据上述支持图像的第一自关联信息和上述查询图像的第二自关联信息确定的;以及,根据至少一个第二预定尺度的特征信息和上述至少一个第一预定尺度的相关信息,得到上述查询图像的图像分割结果。
根据本公开的另一方面,提供了一种图像分割模型的训练方法,包括:根据样本查询图像的样本查询图像数据、样本支持图像的样本支持图像数据和与上述样本支持图像对应的样本掩码图像的样本掩码图像数据,得到至少一个尺度的样本特征信息,其中,上述样本特征信息包括上述样本查询图像的样本查询特征信息、上述样本支持图像的样本支持特征信息和上述样本掩码图像的样本掩码特征信息;根据上述至少一个尺度的样本特征信息,得到至少一个第一预定尺度的样本相关信息,其中,上述样本相关信息包括以下至少一项:上述样本查询图像和上述样本支持图像之间的互相关样本语义信息以及上述样本查询图像和上述样本支持图像之间的互关联样本信息,上述样本查询图像和上述样本支持图像之间的互相关样本语义信息是根据上述样本支持图像的第一自相关样本语义信息和上述样本查询图像的第二自相关样本语义信息确定的,上述样本查询图像和上述样本支持图像之间的互关联样本信息是根据上述样本支持图像的第一自关联样本信息和上述样本查询图像的第二自关联样本信息确定的;根据至少一个第二预定尺度的样本特征信息和上述至少一个第一预定尺度的样本相关信息,得到上述查询图像的图像分割结果;以及,根据上述查询图像的真实分割结果和图像分割结果训练上述图像分割模型,得到训练完成的图像分割模型。
根据本公开的另一方面,提供了一种图像分割装置,包括:第一获得模块,用于根据查询图像的查询图像数据、支持图像的支持图像数据和与上述支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息,其中,上述特征信息包括上述查询图像的查询特征信息、上述支持图像的支持特征信息和上述掩码图像的掩码特征信息;第二获得模块,用于根据上述至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,其中,上述相关信息包括以下至少一项:上述查询图像和上述支持图像之间的互相关语义信息以及上述查询图像和上述支持图像之间的互关联信息,上述互相关语义信息是根据上述支持图像的第一自相关语义信息和上述查询图像的第二自相关语义信息确定的,上述互关联信息是根据上述支持图像的第一自关联信息和上述查询图像的第二自关联信息确定的;以及,第三获得模块,用于根据至少一个第二预定尺度的特征信息和上述至少一个第一预定尺度的相关信息,得到上述查询图像的图像分割结果。
根据本公开的另一方面,提供了一种图像分割模型的训练装置,包括:第四获得模块,用于根据样本查询图像的样本查询图像数据、样本支持图像的样本支持图像数据和与上述样本支持图像对应的样本掩码图像的样本掩码图像数据,得到至少一个尺度的样本特征信息,其中,上述样本特征信息包括上述样本查询图像的样本查询特征信息、上述样本支持图像的样本支持特征信息和上述样本掩码图像的样本掩码特征信息;第五获得模块,用于根据上述至少一个尺度的样本特征信息,得到至少一个第一预定尺度的样本相关信息,其中,上述样本相关信息包括以下至少一项:上述样本查询图像和上述样本支持图像之间的互相关样本语义信息以及上述样本查询图像和上述样本支持图像之间的互关联样本信息,上述样本查询图像和上述样本支持图像之间的互相关样本语义信息是根据上述样本支持图像的第一自相关样本语义信息和上述样本查询图像的第二自相关样本语义信息确定的,上述样本查询图像和上述样本支持图像之间的互关联样本信息是根据上述样本支持图像的第一自关联样本信息和上述样本查询图像的第二自关联样本信息确定的;第六获得模块,用于根据至少一个第二预定尺度的样本特征信息和上述至少一个第一预定尺度的样本相关信息,得到上述样本查询图像的预测图像分割结果;以及,第七获得模块,用于根据上述样本查询图像的真实图像分割结果和预测图像分割结果训练上述图像分割模型,得到训练完成的图像分割模型。
根据本公开的另一方面,提供了一种电子设备,包括:至少一个处理器;以及与上述至少一个处理器通信连接的存储器;其中,上述存储器存储有可被上述至少一个处理器执行的指令,上述指令被上述至少一个处理器执行,以使上述至少一个处理器能够执行本公开所述的方法。
根据本公开的另一方面,提供了一种存储有计算机指令的非瞬时计算机可读存储介质,其中,上述计算机指令用于使上述计算机执行本公开所述的方法。
根据本公开的另一方面,提供了一种计算机程序产品,包括计算机程序,上述计算机程序在被处理器执行时实现本公开所述的方法。
应当理解,本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征,也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
附图用于更好地理解本方案,不构成对本公开的限定。其中:
图1示意性示出了根据本公开实施例的可以应用图像分割方法、图像分割模型的训练方法及装置的示例性系统架构;
图2示意性示出了根据本公开实施例的图像分割方法的流程图;
图3A示意性示出了根据本公开实施例的确定查询图像和支持图像之间的互相关语义向量过程的示例示意图;
图3B示意性示出了根据本公开实施例的确定查询图像和支持图像之间的互关联向量过程的示例示意图;
图3C示意性示出了根据本公开实施例的图像分割过程的示例示意图;
图3D示意性示出了根据本公开另一实施例的图像分割过程的示例示意图;
图3E示意性示出了根据本公开另一实施例的图像分割过程的示例示意图;
图4示意性示出了根据本公开实施例的图像分割模型的训练方法的流程图;
图5示意性示出了根据本公开实施例的图像分割装置的框图;
图6示意性示出了根据本公开实施例的图像分割模型的训练装置的框图;以及
图7示意性示出了根据本公开实施例的适于实现图像分割方法和图像分割模型的训练方法的电子设备的框图。
具体实施方式
以下结合附图对本公开的示范性实施例做出说明,其中包括本公开实施例的各种细节以助于理解,应当将它们认为仅仅是示范性的。因此,本领域普通技术人员应当认识到,可以对这里描述的实施例做出各种改变和修改,而不会背离本公开的范围和精神。同样,为了清楚和简明,以下的描述中省略了对公知功能和结构的描述。
对象可以包括内在特征和外在特征。内在特征可以指不随拍摄条件的变化而变化的特征。外在特征可以指随拍摄条件的变化而变化的特征。由此,在不同拍摄条件的情况下对同一对象进行拍摄得到的图像中对象的外在特征不同。例如,对象是飞机。飞机的内在特征可以包括组成部件和制作材料。飞机的外在特征可以包括飞机形状和飞机颜色。拍摄条件可以包括拍摄角度和光照条件。在不同拍摄条件下对同一飞机进行拍摄得到的图像中飞机的外在特征不同。上述针对同一对象拍摄得到的不同图像所对应的类别相同。即,不同图像是属于同一类别的图像。
在小样本分割中,查询图像和与查询图像对应的支持图像是属于同一类别的图像。查询图像和支持图像中同一对象的特征差异较大,即,查询图像和支持图像之间的类内方差较大,由此,给上下文语义信息传递带来了障碍,从而降低了图像分割结果精度。此外,查询图像和支持图像之间的语义匹配关系不够准确,降低了图像分割精度。
基于上述内容,查询图像和支持图像之间的类内方差较大以及查询图像和支持图像之间的语义匹配关系不够准确中的至少一项的存在会降低图像分割精度。
为此,本公开实施例提出了一种图像分割方案。根据查询图像的查询图像数据、支持图像的支持图像数据和与支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息。特征信息包括查询图像的查询特征信息、支持图像的支持特征信息和掩码图像的掩码特征信息。根据至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息。相关信息包括以下至少一项:查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息。互相关语义信息是根据支持图像的第一自相关语义信息和查询图像的第二自相关语义信息确定的。互关联信息是根据支持图像的第一自关联信息和查询图像的第二自关联信息确定的。根据至少一个第二预定尺度的特征信息和至少一个第一预定尺度的相关信息,得到查询图像的图像分割结果。
根据本公开的实施例,通过聚合查询图像的至少一个尺度的特征信息和支持图像的至少一个尺度的特征信息,得到支持图像的第一自相关语义信息和查询图像的第二自相关语义信息,在此基础上,获得更准确的用于表征跨上下文匹配信息的互相关语义信息,实现了融合支持图像的语义信息来生成更准确的语义匹配关系,进而提高了图像分割精度。此外,互关联信息的确定融合了查询图像自身的自关联信息,因此,使得查询图像和支持图像之间的互关联信息更为准确和可靠,由此,可以有效缓解查询图像和支持图像之间的类内差异,进而提高了图像分割精度。此外,由于相关信息是至少一个第一预定尺度的相关信息,因此,实现了利用支持图像的多尺度互相关语义信息和互关联信息来引导查询图像的图像分割结果。
图1示意性示出了根据本公开实施例的可以应用图像分割方法、图像分割模型的训练方法及装置的示例性系统架构。
需要注意的是,图1所示仅为可以应用本公开实施例的系统架构的示例,以帮助本领域技术人员理解本公开的技术内容,但并不意味着本公开实施例不可以用于其他设备、系统、环境或场景。例如,在另一实施例中,可以应用图像分割方法、图像分割模型的训练方法及装置的示例性系统架构可以包括终端设备,但终端设备可以无需与服务器进行交互,即可实现本公开实施例提供的图像分割方法、图像分割模型的训练方法及装置。
如图1所示,根据该实施例的系统架构100可以包括终端设备101、102、103,网络104和服务器105。网络104用以在终端设备101、102、103和服务器105之间提供通信链路的介质。网络104可以包括各种连接类型,例如有线和/或无线通信链路等。
用户可以使用终端设备101、102、103通过网络104与服务器105交互,以接收或发送消息等。终端设备101、102、103上可以安装有各种通讯客户端应用。例如,知识阅读类应用、网页浏览器应用、搜索类应用、即时通信工具、邮箱客户端或社交平台软件等(仅为示例)。
终端设备101、102、103可以是具有显示屏并且支持网页浏览的各种电子设备,包括但不限于智能手机、平板电脑、膝上型便携计算机和台式计算机等。
服务器105可以是提供各种服务的各种类型的服务器。例如,例如,服务器105可以是云服务器,又称为云计算服务器或云主机,是云计算服务体系中的一项主机产品,以解决了传统物理主机与VPS服务(Virtual Private Server,虚拟专用服务器)中,存在的管理难度大,业务扩展性弱的缺陷。服务器105也可以为分布式系统的服务器,或者是结合了区块链的服务器。
需要说明的是,本公开实施例所提供的图像分割方法一般可以由终端设备101、102、或103执行。相应地,本公开实施例所提供的图像分割装置也可以设置于终端设备101、102、或103中。
备选地,本公开实施例所提供的图像分割方法一般也可以由服务器105执行。相应地,本公开实施例所提供的图像分割装置一般可以设置于服务器105中。本公开实施例所提供的图像分割方法也可以由不同于服务器105且能够与终端设备101、102、103和/或服务器105通信的服务器或服务器集群执行。相应地,本公开实施例所提供的图像分割装置也可以设置于不同于服务器105且能够与终端设备101、102、103和/或服务器105通信的服务器或服务器集群中。
需要说明的是,本公开实施例所提供的图像分割模型的训练方法一般可以由服务器105执行。相应地,本公开实施例所提供的图像分割模型的训练装置一般可以设置于服务器105中。本公开实施例所提供的图像分割模型的训练方法也可以由不同于服务器105且能够与终端设备101、102、103和/或服务器105通信的服务器或服务器集群执行。相应地,本公开实施例所提供的图像分割模型的训练装置也可以设置于不同于服务器105且能够与终端设备101、102、103和/或服务器105通信的服务器或服务器集群中。
备选地,本公开实施例所提供的图像分割方法一般也可以由终端设备101、102、或103执行。相应地,本公开实施例所提供的图像分割装置也可以设置于终端设备101、102、或103中。
应该理解,图1中的终端设备、网络和服务器的数目仅仅是示意性的。根据实现需要,可以具有任意数目的终端设备、网络和服务器。
应注意,以下方法中各个操作的序号仅作为该操作的表示以便描述,而不应被看作表示该各个操作的执行顺序。除非明确指出,否则该方法不需要完全按照所示顺序来执行。
图2示意性示出了根据本公开实施例的图像分割方法的流程图。
如图2所示,该方法200包括操作S210~S230。
在操作S210,根据查询图像的查询图像数据、支持图像的支持图像数据和与支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息。特征信息包括查询图像的查询特征信息、支持图像的支持特征信息和掩码图像的掩码特征信息。
在操作S220,根据至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息。相关信息包括以下至少一项:查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息。互相关语义信息是根据支持图像的第一自相关语义信息和查询图像的第二自相关语义信息确定的。互关联信息是根据支持图像的第一自关联信息和查询图像的第二自关联信息确定的。
在操作S230,根据至少一个第二预定尺度的特征信息和至少一个第一预定尺度的相关信息,得到查询图像的图像分割结果。
根据本公开的实施例,查询图像(即Query Image)可以指需要进行图像分割的图像。支持图像(即Support Image)可以对查询图像的图像分割结果起到引导作用。查询图像可以包括至少一个对象。每个对象具有与该对象对应的类别。由此,查询图像的类别可以包括至少一个。针对查询图像包括的每个类别,可以具有与该类别对应的至少一个支持图像。与该类别对应的支持图像包括与该类别对应的对象。即,可以根据查询图像包括的类别,确定与查询图像对应的支持图像。支持图像的数目可以包括一个或多个。具有与支持图像对应的掩码图像(即Mask Image)。掩码图像可以指支持图像的图像分割结果。
根据本公开的实施例,自相关语义信息可以指与图像自身相关的语义信息。自相关语义信息还可以称为上下文语义信息。互相关语义信息可以指不同图像之间的语义信息。互相关语义信息可以用于构建查询图像和样本图像之间的语义匹配关系。互相关信息还可以称为跨上下文语义匹配信息。
根据本公开的实施例,第一自相关语义信息可以指与支持图像自身相关的语义信息。第二自相关语义信息可以指与查询图像自身相关的语义信息。查询图像和支持图像之间的互相关语义信息可以是根据第一自相关语义信息和第二自相关语义信息确定的。
根据本公开的实施例,自关联信息可以指图像自身的关联信息。互关联信息可以指不同图像之间的关联信息。互相关信息可以用于支持支持图像对查询图像的细节信息的引导。
根据本公开的实施例,第一自关联信息可以指支持图像自身的关联信息。第二自关联信息可以指查询图像自身的关联信息。查询图像和支持图像之间的互关联信息可以是根据第一自关联信息和第二自关联信息确定的。由于互关联信息是根据查询图像的第一自关联信息确定的,即,融合了查询图像自身的自关联信息,因此,使得查询图像和支持图像之间的互关联信息更为准确和可靠,由此,可以有效缓解查询图像和支持图像之间的类内差异。
根据本公开的实施例,可以对查询图像数据、支持图像数据和掩码图像数据进行特征提取,得到至少一个尺度的与查询图像对应的查询特征信息、至少一个尺度的与支持图像对应的支持特征信息和至少一个尺度的与掩码图像对应的掩码特征信息。特征信息可以包括特征向量。查询特征信息可以包括查询特征向量。支持特征信息可以包括支持特征向量。掩码特征信息可以包括掩码特征向量。
根据本公开的实施例,可以利用图像分割模型的特征提取模块处理查询图像数据、支持图像数据和掩码图像数据,得到至少一个尺度的程序特征信息、至少一个尺度的支持特征信息和至少一个尺度的掩码特征信息。特征提取模块可以包括编码器或卷积神经网络。编码器可以包括基于Transformer的编码器。Transformer可以包括视觉Transformer。视觉Transformer可以包括Vision Transformer或Swin Transformer。卷积神经网络可以包括ResNet(Residual Neural Network,残差神经网络)、VGGNet(Visual Geometry GroupNetwork,视觉几何群网络)、WideResNet(Wide Residual Network,宽残差网络)或DenseNet(Dense Neural Network,稠密神经网络)。
根据本公开的实施例,特征提取模块可以包括第一特征提取子模块、第二特征提取子模块和第三特征提取子模块。第一特征提取子模块可以用于提取支持图像的特征信息。第二特征提取子模块可以用于提取查询图像的特征信息。第三特征提取子模块可以用于提取掩码图像的特征信息。第一特征提取子模块可以包括至少一个级联的第一特征提取单元。第二特征提取子模块可以包括至少一个级联的第二特征提取单元。第三特征提取子模块可以包括至少一个级联的第三特征提取单元。第一特征提取子模块可以是第一编码器(即第一Image Encoder)。第二特征提取子模块可以是第二编码器(即第二ImageEncoder)。第三特征提取子模块可以是第三编码器(即Mask Encoder)。第一特征提取子模块、第二特征提取子模块和第三特征提取子模块的模型结构可以相同或不同。图像的图像数据依次经由至少一个级联的特征提取单元,每通过一个特征提取单元得到的与图像特征信息对应的特征图的尺寸减小。每个特征提取单元可以用于实现对与该特征提取单元对应的尺度的特征信息进行下采样。
根据本公开的实施例,可以利用第一特征提取子模块处理支持图像数据,得到至少一个尺度的支持特征信息。利用第二特征提取子模块处理查询图像数据,得到至少一个尺度的查询特征信息。利用第三特征提取子模块处理掩码图像数据,得到至少一个尺度的掩码特征信息。
根据本公开的实施例,第一预定尺度可以是至少一个尺度中的部分尺度或全部尺度。第二预定尺度也可以是至少一个尺度中的部分尺度或全部尺度。第一预定尺度与第二预定尺度可以相同或不同。例如,至少一个尺度可以包括四个尺度,即,第一尺度、第二尺度、第三尺度和第四尺度。至少一个第一预定尺度可以包括四个尺度中的后两个尺度,即,第三尺度和第四尺度。至少一个第二预定尺度可以包括四个尺度中的前两个尺度,即,第一尺度和第二尺度。
根据本公开的实施例,根据至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息可以包括:根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互相关语义信息。备选地,根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互关联信息。备选地,根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互相关语义信息以及至少一个第一预定尺度的查询图像和支持图像之间的互关联信息。
根据本公开的实施例,根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互相关语义信息,可以包括:根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像的第二自相关语义信息和至少一个第一预定尺度的支持图像的第一自相关语义信息。根据至少一个第一预定尺度的第一自相关语义信息和至少一个第一预定尺度的第二自相关语义信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互相关语义信息。第一自相关语义信息可以包括第一自相关语义向量。第二自相关语义信息可以包括第二自相关语义向量。互相关语义信息可以包括互相关语义向量。
根据本公开的实施例,根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互关联信息,可以包括:根据至少一个尺度的特征信息,得到至少一个第一预定尺度的查询图像的第二自关联信息和至少一个第一预定尺度的第一自关联信息。根据至少一个第一预定尺度的第一自关联信息和至少一个第一预定尺度的第二自关联信息,得到至少一个第一预定尺度的查询图像和支持图像之间的互关联信息。第一自关联信息可以包括第一自关联向量。第二自关联信息可以包括第二自关联向量。互关联信息可以包括互关联向量。
根据本公开的实施例,可以利用解码器处理至少一个第二预定尺度的特征信息和至少一个第一预定尺度的相关信息,得到查询图像的图像分割结果。
根据本公开的实施例,通过聚合查询图像的至少一个尺度的特征信息和支持图像的至少一个尺度的特征信息,得到支持图像的第一自相关语义信息和查询图像的第二自相关语义信息,在此基础上,获得更准确的用于表征跨上下文匹配信息的互相关语义信息,实现了融合支持图像的语义信息来生成更准确的语义匹配关系,进而提高了图像分割精度。此外,互关联信息的确定融合了查询图像自身的自关联信息,因此,使得查询图像和支持图像之间的互关联信息更为准确和可靠,由此,可以有效缓解查询图像和支持图像之间的类内差异,进而提高了图像分割精度。此外,由于相关信息是至少一个第一预定尺度的相关信息,因此,实现了利用支持图像的多尺度互相关语义信息和互关联信息来引导查询图像的图像分割结果。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息的情况下,操作S220可以包括如下操作。
利用至少一个关系引导的上下文转换器(Relation-guided ContextTransformer,RCT)处理至少一个尺度的特征信息,得到至少一个第一预定尺度的互相关语义信息。
根据本公开的实施例,针对至少一个预定尺度中的每个预定尺度,可以具有与该预定尺度对应的关系引导的上下文转换器。关系引导的上下文转换器可以用于根据至少一个尺度的特征信息确定与该预定尺度对应的互相关语义信息。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息的情况下,操作S220可以包括如下操作。
针对至少一个第一预定尺度中的每个第一预定尺度,根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一自相关语义信息。根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二自相关语义信息。根据与第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与第一预定尺度对应的互相关语义信息。
根据本公开的实施例,可以针对至少一个第一预定尺度中的每个第一预定尺度,可以对与第一预定尺度对应的支持特征信息和掩码特征信息进行处理,得到第一自相关语义信息。对与第一预定尺度对应的查询特征信息进行处理,得到第二自相关语义信息。对与第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息进行处理,得到互相关语义信息。第一自相关语义信息可以包括第一自相关语义向量。第二自相关语义信息可以包括第二自相关语义向量。互相关语义信息可以包括互相关语义向量。
根据本公开的实施例,根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一自相关语义信息,可以包括如下操作。
基于第一自注意力策略,根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一中间信息。根据与第一预定尺度对应的第一中间信息和掩码特征信息,得到与第一预定尺度对应的第一自相关语义信息。
根据本公开的实施例,自注意力策略可以用于实现以高权重去聚焦重要信息,低权重去忽略非重要信息,并能通过共享重要信息与其他信息进行信息交换,从而实现重要信息的传递。
根据本公开的实施例,可以基于第一自注意力策略确定第一自注意力层。第一自注意力层可以包括第一多头自注意力(Multi-Head Self Attention,MHA)层。利用第一自注意力层处理与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一中间信息。第一中间信息可以包括第一中间向量。
根据本公开的实施例,在确定与第一预定尺度对应的第一中间信息之后,可以利用第一前馈神经网络(Feed Forward Network,FFN)层处理与第一预定尺度对应的第一中间信息和掩码特征信息,得到与第一预定尺度对应的第一自相关语义信息。前馈神经网络层还可以被称为多层感知机层(MultiLayer Perceptron,MLP)。
根据本公开的实施例,可以将与第一预定尺度对应的第一中间信息和掩码特征信息进行融合操作,得到第一融合信息。利用第一前馈神经网络层处理与第一预定尺度对应的第一融合信息,得到与第一预定尺度对应的第一自相关语义信息。例如,可以将与第一预定尺度对应的第一中间向量和掩码特征向量进行融合,得到第一融合向量。利用第一前馈神经网络层处理与第一预定尺度对应的第一融合向量,得到与第一预定尺度对应的第一自相关语义向量。融合操作可以包括以下一项:相加操作、拼接操作和连接(即Concatenate)操作。
根据本公开的实施例,根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二自相关语义信息,可以包括如下操作。
基于第二自注意力策略,根据与第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二中间信息。根据与第一预定尺度对应的第二中间信息和查询特征信息,得到与第一预定尺度对应的第二自相关语义信息。
根据本公开的实施例,可以基于第二自注意力策略确定第二自注意力层。第二自注意力层可以包括第二多头自注意力层。利用第二自注意力层处理与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二中间信息。第二中间信息可以包括第二中间向量。
根据本公开的实施例,在确定与第一预定尺度对应的第二中间信息之后,可以利用第二前馈神经网络层处理与第一预定尺度对应的第二中间信息和查询特征信息,得到与第一预定尺度对应的第二自相关语义信息。
根据本公开的实施例,可以将与第一预定尺度对应的第二中间信息和查询特征信息进行融合操作,得到第二融合信息。利用第二前馈神经网络层处理与第一预定尺度对应的第二融合信息,得到与第一预定尺度对应的第二自相关语义信息。例如,可以将与第一预定尺度对应的第二中间向量和查询特征向量进行融合,得到第二融合向量。利用第二前馈神经网络层处理与第一预定尺度对应的第二融合向量,得到与第一预定尺度对应的第二自相关语义向量。融合操作可以包括以下一项:相加操作、拼接操作和连接操作。
根据本公开的实施例,根据与第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与第一预定尺度对应的互相关语义信息,可以包括如下操作。
基于第三自注意力策略,根据与第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与第一预定尺度对应的第三中间信息。根据与第一预定尺度对应的第一自相关语义信息和第三中间信息,得到与第一预定尺度对应的互相关语义信息。
根据本公开的实施例,可以基于第三自注意力策略确定第三自注意力层。第三自注意力层可以包括第三多头自注意力层。利用第三自注意力层处理与第一预定尺度对应的第一自相关语义信息、第二自相关语义信息和支持特征信息,得到与第一预定尺度对应的第三中间信息。第三中间信息可以包括第三中间向量。
根据本公开的实施例,在确定与第一预定尺度对应的第三中间信息之后,可以利用第三前馈神经网络层处理与第一预定尺度对应的第三中间信息和第一自相关语义信息,得到与第一预定尺度对应的互相关语义信息。
根据本公开的实施例,可以将与第一预定尺度对应的第三中间信息和第一自相关语义信息进行融合操作,得到第三融合信息。利用第三前馈神经网络层处理与第一预定尺度对应的第三融合信息,得到与第一预定尺度对应的互相关语义信息。例如,可以将与第一预定尺度对应的第三中间向量和第一自相关语义向量进行融合,得到第三融合向量。利用第三前馈神经网络层处理与第一预定尺度对应的第三融合向量,得到与第一预定尺度对应的互相关语义向量。融合操作可以包括以下一项:相加操作、拼接操作和连接操作。
根据本公开的实施例,关系引导的上下文转换器可以包括第一多头自注意力层、第一前馈神经网络层、第二多头自注意力层、第二前馈神经网络层、第三多头自注意层和第三前馈神经网络层。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互关联信息的情况下,操作S220可以包括如下操作。
根据本公开的实施例,利用至少一个关系引导的关联转换器(Relation-guidedAffinity Transformer,RAT)处理至少一个尺度的特征信息,得到至少一个第一预定尺度的互关联信息。
根据本公开的实施例,针对至少一个预定尺度中的每个预定尺度,可以具有与该预定尺度对应的关系引导的关联转换器。关系引导的关联转换器可以用于根据至少一个尺度的特征信息确定与该预定尺度对应的互关联信息。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互关联信息的情况下,操作S220可以包括如下操作。
针对至少一个第一预定尺度中的每个第一预定尺度,根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一自关联信息。根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二自关联信息。根据与第一预定尺度对应的支持特征信息和查询特征信息,得到与第一预定尺度对的中间互关联信息。根据与第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息,得到与第一预定尺度对应的互关联信息。
根据本公开的实施例,可以针对至少一个第一预定尺度中的每个第一预定尺度,可以对与第一预定尺度对应的支持特征信息和掩码特征信息进行处理,得到第一自关联信息。对与第一预定尺度对应的查询特征信息进行处理,得到第二自相关语义信息。对与第一预定尺度对应的支持特征信息和查询特征信息进行处理,得到中间互关联信息。再对与第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息进行处理,得到互关联信息。第一自关联信息可以包括第一自关联向量。第二自关联信息可以包括第二自关联向量。中间互关联信息可以包括中间互关联向量。互关联信息可以包括互关联向量。
根据本公开的实施例,根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一自关联信息,可以包括如下操作。
对与第一预定尺度对应的支持特征信息和掩码特征信息进行点积操作,得到与第一预定尺度对应的第四中间信息。对与第一预定尺度对应的第四中间信息进行全连接操作,得到与第一预定尺度对应的第一自关联信息。
根据本公开的实施例,对与第一预定尺度对应的第四中间信息进行全连接操作,得到与第一预定尺度对应的第一自关联信息,可以包括:可以对与第一预定尺度对应的第四中间信息进行全连接操作,得到与第一预定尺度对应的第七中间信息。对与第一预定尺度对应的第四中间信息进行全连接操作,得到与第一预定尺度对应的第八中间信息。可以对与第一预定尺度对应的第七中间信息和第八中间信息进行连接操作,得到与第一预定尺度对应的第一自关联信息。第四中间信息可以包括第四中间向量。第七中间信息可以包括第七中间向量。第八中间信息可以包括第八中间向量。
例如,可以利用第一全连接(Fully Connected,FC)层处理与第一预定尺度对应的第四中间信息,得到与第一预定尺度对应的第七中间信息。可以利用第二全连接层处理与第一预定尺度对应的第四中间信息,得到与第一预定尺度对应的第八中间信息。
根据本公开的实施例,根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二自关联信息,可以包括如下操作。
对与第一预定尺度对应的查询特征信息进行全连接操作,得到与第一预定尺度对应的第二自关联信息。
根据本公开的实施例,可以对与第一预定尺度对应的查询特征信息进行全连接操作,得到与第一预定尺度对应的第九中间信息。对与第一预定尺度对应的查询信息进行全连接操作,得到与第一预定尺度对应的第十中间信息。可以对与第一预定尺度对应的第九中间信息和第十中间信息进行连接操作,得到与第一预定尺度对应的第二自关联信息。第九中间信息可以包括第九中间向量。第十中间信息可以包括第十中间向量。
例如,可以利用第三全连接层处理与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第九中间信息。可以利用第四全连接层处理与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第十中间信息。
根据本公开的实施例,根据与第一预定尺度对应的支持特征信息和查询特征信息,得到与第一预定尺度对的中间互关联信息,可以包括如下操作。
对与第一预定尺度对应的支持特征信息和查询特征信息进行全连接操作,得到与第一预定尺度对应的中间互关联信息。
根据本公开的实施例,可以对与第一预定尺度对应的支持特征信息进行全连接操作,得到与第一预定尺度对应的第十一中间信息。对与第一预定尺度对应的查询信息进行全连接操作,得到与第一预定尺度对应的第十二中间信息。可以对与第一预定尺度对应的第十一中间信息和第十二中间信息进行连接操作,得到与第一预定尺度对应的中间互关联信息。第十一中间信息可以包括第十一中间向量。第十二中间信息可以包括第十二中间向量。
例如,可以利用第五全连接层处理与第一预定尺度对应的支持特征信息,得到与第一预定尺度对应的第十一中间信息。可以利用第六全连接层处理与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第十二中间信息。
根据本公开的实施例,根据与第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息,得到与第一预定尺度对应的互关联信息,可以包括如下操作。
基于第四自注意力策略,根据与第一预定尺度对应的第一自关联信息和中间互关联信息,得到与第一预定尺度对应的第五中间信息。根据与第一预定尺度对应的第二自关联信息和第五中间信息,得到与第一预定尺度对应的互关联信息。
根据本公开的实施例,可以基于第四自注意力策略确定第四自注意力层。第四自注意力层可以包括第四多头自注意力层。利用第四自注意力层处理与第一预定尺度对应的第一自关联信息和中间互关联信息,得到与第一预定尺度对应的第五中间信息。第五中间信息可以包括第五中间向量。
根据本公开的实施例,在确定与第一预定尺度对应的第五中间信息之后,可以利用第四前馈神经网络层处理与第一预定尺度对应的第五中间信息和第二自关联信息,得到与第一预定尺度对应的互关联信息。
根据本公开的实施例,可以将与第一预定尺度对应的第五中间信息和第二自关联信息进行融合操作,得到第四融合信息。利用第四前馈神经网络层处理与第一预定尺度对应的第四融合信息,得到与第一预定尺度对应的互关联信息。例如,可以将与第一预定尺度对应的第五中间向量和第二自关联向量进行融合操作,得到第四融合向量。利用第四前馈神经网络层处理与第一预定尺度对应的第四融合向量,得到与第一预定尺度对应的互关联向量。融合操作可以包括以下一项:相加操作、拼接操作和连接操作。
根据本公开的实施例,关系引导的关联转换器可以包括第一全连接层、第二全连接层、第三全连接层、第四全连接层、第五全连接层、第六全连接层、第四多头自注意层和第四前馈神经网络层。
根据本公开的实施例,根据至少一个第二预定尺度的特征信息和至少一个第一预定尺度的相关信息,得到查询图像的图像分割结果,可以包括如下操作。
根据至少一个第一预定尺度的相关信息,得到第六中间信息。根据第六中间信息和至少一个第二预定尺度的查询特征信息,得到查询图像的图像分割结果。
根据本公开的实施例,在确定相关信息包括查询图像和支持图像之间的互相关语义信息的情况下,可以利用图像分割模型包括的解码器(即Decoder)中与至少一个预定尺度对应的至少一个解码单元处理至少一个第一预定尺度的互相关语义信息,得到第六中间信息。第六中间信息可以包括第六中间向量。可以利用解码器中与至少一个第二预定尺度对应的至少一个解码单元处理第六中间信息和至少一个第二预定尺度的查询特征信息,得到查询图像的图像分割结果。解码器可以包括包括至少一个级联的解码单元。每个解码单元可以用于实现对与该解码单元对应的尺度的特征信息进行上采样。
根据本公开的实施例,至少一个尺度可以包括N个尺度。至少一个第一预定尺度可以包括M个第一预定尺度。至少一个第二预定尺度可以包括(N-M)个第二预定尺度。N可以是大于1的整数。M可以是大于或等于1且小于N的整数。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息以及查询图像和所述支持图像之间的互关联信息中的一项的情况下,根据至少一个第一预定尺度的相关信息,得到第六中间信息,可以包括如下操作。
在(N-M+1)≤i<N的情况下,对与第i个第一预定尺度对应的相关信息进行卷积操作,得到与第i个第一预定尺度对应的第一原始中间信息。对与第i个第一预定尺度对应的第一原始中间信息和与第(i+1)个第一预定尺度对应的第一目标中间信息进行点积操作,得到与第i个第一预定尺度对应的第一目标中间信息。将与第(N-M+1)个第一预定尺度对应的第一目标中间信息确定为第六中间信息。
根据本公开的实施例,N和M的数值均可以根据实际业务需要进行配置,在此不作限定。例如,N=4。M=2。i∈{N,(N-1),......,(N-M+2),(N-M+1)}。第一原始中间信息可以包括第一原始中间向量。第一目标中间信息可以包括第一目标中间向量。
根据本公开的实施例,在i=N的情况下,对与第N个第一预定尺度对应的相关信息进行卷积操作,得到与第N个第一预定尺度对应的第一目标中间信息。例如,可以利用与第N个第一预定尺度对应的第一卷积层处理与第N个第一预定尺度对应的相关信息,得到与第N个第一预定尺度对应的第一目标中间信息。
根据本公开的实施例,在(N-M+1)≤i<N的情况下,可以利用与第i个第一预定尺度对应的第一卷积层处理与第i个第一预定尺度对应的相关信息,得到与第i个第一预定尺度对应的第一原始中间信息。
根据本公开的实施例,可以根据如下公式(1)确定与第一预定尺度对应的第一自相关语义向量。
根据本公开的实施例,表征与第i个第一预定尺度对应的第一自相关语义向量。表征与第i个第一预定尺度对应的支持特征向量。表征与第i个第一预定尺度对应的掩码特征向量。LN()表征层次归一化(即LayerNorm)。
根据本公开的实施例,可以根据如下公式(2)确定与第一预定尺度对应的第二自相关语义向量。
根据本公开的实施例,和均具有自身语义信息。可以用于构建像素级的跨上下文信息。基于专注于前景对象,但背景信息也有利于构建语义匹配关系,因此,可以根据如下公式(3)确定与第一预定尺度对应的互相关语义向量。
根据本公开的实施例,可以根据如下公式(4)确定与第一预定尺度对应的第一自关联向量。
根据本公开的实施例,表征与第i个第一预定尺度对应的第一自关联向量。Θ表征通道维度的连接操作。softmax()表征用于注意力机制的归一化函数。表征与第i个第一预定尺度对应的查询特征向量。Wq和Wk均表征权重矩阵。表征与第i个第一预定尺度对应的掩码特征图的通道数目。Ci表征与第i个第一预定尺度对应的通道数目。
根据本公开的实施例,可以根据如下公式(5)确定与第一预定尺度对应的第二自关联向量。
根据本公开的实施例,可以根据如下公式(6)确定与第一预定尺度对应的中间自关联向量。
根据本公开的实施例,可以根据如下公式(7)确定与第一预定尺度对应的自关联向量。
根据本公开的实施例,根据第六中间信息和至少一个第二预定尺度的查询特征信息,得到查询图像的图像分割结果,可以包括如下操作。
在j=N-M的情况下,对与第(N-M)个第二预定尺度对应的查询特征信息进行卷积操作,得到与第(N-M)个第二预定尺度对应的第一原始中间信息。对第六中间信息和与第(N-M)个第二预定尺度对应的第一原始中间信息进行点积操作,得到与第(N-M)个第二预定尺度对应的第一目标中间信息。在1≤j<N-M的情况下,对与第j个第二预定尺度对应的查询特征信息进行卷积操作,得到与第j个第二预定尺度对应的第一原始中间信息。对与第j个第二预定尺度对应的第一原始中间信息和与第(j+1)个第二预定尺度对应的第一目标中间信息进行点积操作,得到与第j个第二预定尺度对应的第一目标中间信息。根据与第1个第二预定尺度对应的第一目标中间信息,确定查询图像的图像分割结果。
根据本公开的实施例,j∈{(N-M),(N-M-1),......,2,1}。
根据本公开的实施例,在j=N-M的情况下,可以利用与第(N-M)个第二预定尺度对应的第一卷积层处理与第(N-M)个第二预定尺度对应的查询特征信息,得到与第(N-M)个第二预定尺度对应的第一原始中间信息。
根据本公开的实施例,在1≤j<N-M的情况下,可以利用与第j个第二预定尺度对应的第一卷积层处理与第j个第二预定尺度对应的查询特征信息,得到与第j个第二预定尺度对应的第一原始中间信息。
根据本公开的实施例,可以对与第1个第二预定尺度对应的第一目标中间信息进行卷积操作,得到查询图像的图像分割结果。
根据本公开的实施例,与后M个尺度的特征信息对应的特征图的尺寸较小,因此,对后M个尺度的特征信息进行处理得到与后M个尺度各自对应的相关信息,可以降低数据处理量。并且,使得后M个尺度的特征信息包括高层次信息,由此,有助于提高图像分割精度。高层次信息可以包括高层次关联信息和高层次语义信息中的至少一项。与前(N-M)个尺度的特征信息对应的特征图的尺寸较大,因此,特征信息包括更多的浅层信息,由此,有助于提高图像分割精度。浅层信息可以包括细节信息。
根据本公开的实施例,至少一个尺度可以包括P个尺度。至少一个第一预定尺度可以包括Q个第一预定尺度。至少一个第二预定尺度可以包括(P-Q)个第二预定尺度。P可以是大于1的整数。Q可以是大于或等于1且小于P的整数。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息的情况下,根据至少一个第一预定尺度的相关信息,得到第六中间信息,可以包括如下操作。
在(P-Q+1)≤k<P的情况下,对与第k个第一预定尺度对应的互相关语义信息和互关联信息进行点积操作,得到与第k个第一预定尺度对应的第二原始中间信息。对与第k个第一预定尺度对应的第二原始中间信息进行卷积操作,得到与第k个第一预定尺度对应的第三原始中间信息。对与第k个第一预定尺度对应的第三原始中间信息和与第(k+1)个第一预定尺度对应的第二目标中间信息进行点积操作,得到与第k个第一预定尺度对应的第二目标中间信息。将与第(P-Q+1)个第一预定尺度对应的第二目标中间信息确定为第六中间信息。
根据本公开的实施例,P和Q的数值均可以根据实际业务需要进行配置,在此不作限定。例如,P=4。Q=2。k∈{P,(P-1),......,(P-Q+2),(P-Q+1)}。第二原始中间信息可以包括第二原始中间向量。第三原始中间信息可以包括第三原始中间向量。第二目标中间信息可以包括第二目标中间向量。
根据本公开的实施例,在k=P的情况下,对与第P个第一预定尺度对应的互相关语义信息和互关联语义信息进行点积操作,得到与第P个第一预定尺度对应的第二原始中间信息。对与第P个第一预定尺度对应的第二原始中间信息进行卷积操作,得到与第P个第一预定尺度对应的第二目标中间信息。例如,可以利用与第P个第一预定尺度对应的第二卷积层处理与第P个第一预定尺度对应的第二原始中间信息,得到与第P个第一预定尺度对应的第二目标中间信息。
根据本公开的实施例,在(P-Q+1)≤k<P的情况下,可以利用与第k个第一预定尺度对应的第二卷积层处理与第k个第一预定尺度对应的第二原始中间信息,得到与第k个第一预定尺度对应的第三原始中间信息。
根据本公开的实施例,根据第六中间信息和至少一个第二预定尺度的查询特征信息,得到查询图像的图像分割结果,可以包括如下操作。
在l=P-Q的情况下,对与第(P-Q)个第二预定尺度对应的查询特征信息进行卷积操作,得到与第(P-Q)个第二预定尺度对应的第二原始中间信息。对第六中间信息和与第(P-Q)个第二预定尺度对应的第二原始中间信息进行点积操作,得到与第(P-Q)个第二预定尺度对应的第二目标中间信息。在1≤l<P-Q的情况下,对与第l个第二预定尺度对应的查询特征信息进行卷积操作,得到与第l个第二预定尺度对应的第二原始中间信息。对与第l个第二预定尺度对应的第二原始中间信息和与第(l+1)个第二预定尺度对应的第二目标中间信息进行点积操作,得到与第l个第二预定尺度对应的第二目标中间信息。根据与第1个第二预定尺度对应的第二目标中间信息,确定查询图像的图像分割结果。
根据本公开的实施例,l∈{(P-Q),(P-Q-1),.....,2,1}。
根据本公开的实施例,在l=P-Q的情况下,可以利用与第(P-Q)个第二预定尺度对应的第二卷积层处理与第(P-Q)个第二预定尺度对应的查询特征信息,得到与第(P-Q)个第二预定尺度对应的第二原始中间信息。
根据本公开的实施例,在1≤l<P-Q的情况下,可以利用与第l个第二预定尺度对应的第二卷积层处理与第l个第二预定尺度对应的查询特征信息,得到与第l个第二预定尺度对应的第二原始中间信息。
根据本公开的实施例,可以对与第1个第二预定尺度对应的第二目标中间信息进行卷积操作,得到查询图像的图像分割结果。
下面参考图3A、图3B、图3C、图3D和图3E,结合具体实施例对根据本公开实施例所述的图像分割方法做进一步说明。
图3A示意性示出了根据本公开实施例的确定查询图像和支持图像之间的互相关语义向量过程的示例示意图。
如图3A所示,在300A中,可以利用关系引导的上下文转换器310处理与第i个第一预定尺度对应的特征向量,得到与第i个第一预定尺度对应的第一自相关语义向量与第i个第一预定尺度对应的第二自相关语义向量和与第i个第一预定尺度对应的互相关语义向量与第i个第一预定尺度对应的特征向量包括与第i个第一预定尺度对应的支持特征向量与第i个第一预定尺度对应的查询特征向量和与第i个第一预定尺度对应的掩码特征向量
根据本公开的实施例,关系引导的上下文转换器310可以包括第一多头自注意力层311、第一前馈神经网络层312、第二多头自注意力层313、第二前馈神经网络层314、第三多头自注意力层315和第三前馈神经网络层316。
图3B示意性示出了根据本公开实施例的确定查询图像和支持图像之间的互关联向量过程的示例示意图。
如图3B所示,在300B中,可以利用关系引导的关联转换器320处理与第i个第一预定尺度对应的特征向量,得到与第i个第一预定尺度对应的第一自关联向量与第i个第一预定尺度对应的第二自关联向量与第i个第一预定尺度对应的中间互关联向量和与第i个第一预定尺度对应的互关联向量与第i个第一预定尺度对应的特征向量包括与第i个第一预定尺度对应的支持特征向量与第i个第一预定尺度对应的查询特征向量和与第i个第一预定尺度对应的掩码特征向量
根据本公开的实施例,关系引导的关联转换器320可以包括第四多头自注意力层321、第四前馈神经网络层322、第一FC(即全连接)层、第二FC层、第三FC层、第四FC层、第五FC层和第六FC层。
根据本公开的实施例,可以对和进行点积操作,得到与第i个第一预定尺度的第四中间向量。利用第一FC层处理与第i个第一预定尺度的第四中间向量,得到与第i个第一预定尺度的第七中间向量。利用第二FC层处理与第i个第一预定尺度的第四中间向量,得到与第i个第一预定尺度的第八中间向量。对与第i个第一预定尺度的第七中间向量和第八中间向量进行连接操作,得到
根据本公开的实施例,可以利用第三FC层处理得到与第i个第一预定尺度的第九中间向量。利用第四FC层处理得到与第i个第一预定尺度的第十中间向量。对与第i个第一预定尺度的第九中间向量和第十中间向量进行连接操作,得到
根据本公开的实施例,可以利用第五FC层处理得到与第i个第一预定尺度的第十一中间向量。利用第六FC层处理得到与第i个第一预定尺度的第十二中间向量。对与第i个第一预定尺度的第十一中间向量和第十二中间向量进行连接操作,得到
图3C示意性示出了根据本公开实施例的图像分割过程的示例示意图。
如图3C所示,在300C中,图像分割模型300可以包括RCT310_1、RCT310_2、第一编码器330、第二编码器340、第三编码器350和解码器360。
根据本公开的实施例,RCT310_1和RCT310_2的模型结构可以与图3A中的RCT310的模型结构相同。N=4。M=2。第一编码器330可以包括四个尺度的特征提取单元,即,S1、S2、S3和S4。第二编码器340可以包括四个尺度的特征提取单元,即,S1、S2、S3和S4。第一编码器330和第二编码器340包括的特征提取单元的模型结构可以相同。第三编码器350可以包括四个尺度的卷积神经网络层。解码器360可以包括四个尺度的卷积神经网络层。双线性上采样被用来保持不同尺度的特征分辨率一致。卷积操作保证不同尺度的特征通道数目一致,可以利用连接操作来级联不同尺度的向量。
根据本公开的实施例,可以利用与第4个第一预定尺度对应的第一卷积层处理得到与第4个第一预定尺度对应的第一目标中间向量。对与第3个第一预定尺度对应的第一卷积层处理得到与第3个第一预定尺度对应的第一原始中间信息。
根据本公开的实施例,可以对与第3个第一预定尺度对应的第一原始中间信息和与第3个第一预定尺度对应的第一目标中间信息进行点积操作,得到与第3个第一预定尺度对应的第一目标中间信息。将与第3个第一预定尺度对应的第一目标中间信息确定为第六中间信息。
根据本公开的实施例,可以利用与第2个第二预定尺度对应的第一卷积层处理得到与第2个第二预定尺度对应的第一原始中间向量。对第六中间信息和与第2个第二预定尺度对应的第一原始中间向量进行点积操作,得到与第2个第二预定尺度对应的第一目标中间向量。
根据本公开的实施例,可以利用第1个第二预定尺度对应的第一卷积层处理得到与第1个第二预定尺度对应的第一原始中间向量。对与第1个第二预定尺度对应的第一原始中间向量和与第2个第二预定尺度对应的第一目标中间信息进行点积操作,得到与第1个第二预定尺度对应的第一目标中间向量。对与第1个第二预定尺度对应的第一目标中间向量进行卷积操作,得到查询图像的图像分割结果304。例如,可以1×1的卷积核和双线性上采样处理第一目标中间向量,得到图像分割结果304。
图3D示意性示出了根据本公开另一实施例的图像分割过程的示例示意图。
如图3D所示,在300D中,图像分割模型300可以包括RAT320_1、RAT320_2、第一编码器330、第二编码器340、第三编码器350和解码器360。
根据本公开的实施例,RAT320_1和RAT320_2的模型结构可以与图3B中的RAT320的模型结构相同。可以利用RAT320_1处理支持特征向量查询特征向量和掩码特征向量得到互关联向量可以利用RAT320_2处理支持特征向量查询特征向量和掩码特征向量得到互关联向量可以将与图3C对应的实施例中的替换为以及替换为其他部分可以参见与图3C对应的实施例相关部分内容的说明,在此不再赘述。
图3E示意性示出了根据本公开另一实施例的图像分割过程的示例示意图。
如图3E所示,在300E中,图像分割模型300可以包括RCT310_1、RCT310_2、RAT320_1、RAT320_2、第一编码器330、第二编码器340、第三编码器350和解码器360。
根据本公开的实施例,RCT310_1和RCT310_2的模型结构可以与图3A中的RCT310的模型结构相同。RAT320_1和RAT320_2的模型结构可以与图3B中的RAT320的模型结构相同。
根据本公开的实施例,可以对和进行点积操作,得到与第4个第一预定尺度对应的第二原始中间向量。可以利用与第4个第一预定尺度对应的第二卷积层处理与第4个第一预定尺度对应的第二原始中间向量,得到与第4个第一预定尺度对应的第二目标中间向量。
根据本公开的实施例,可以对和进行点积操作,得到与第3个第一预定尺度对应的第二原始中间向量。利用与第3个第一预定尺度对应的第二卷积层处理与第3个第一预定尺度对应的第二原始中间向量,得到与第3个第一预定尺度对应的第三原始中间向量。对与第3个第一预定尺度对应的第三原始中间向量和与第4个第一预定尺度对应的第二目标中间向量进行点积操作,得到与第3个第一预定尺度对应的第二目标中间向量。将与第3个第一预定尺度对应的第二目标中间向量确定为第六中间向量。
根据本公开的实施例,可以利用与第2个第二预定尺度对应的第二卷积层处理得到与第2个第二预定尺度对应的第二原始中间向量。对第六中间向量和与第2个第二预定尺度对应的第二原始中间向量进行点积操作,得到与第2个第二预定尺度对应的第二目标中间向量。
根据本公开的实施例,可以利用与第1个第二预定尺度对应的第二卷积层处理得到与第1个第二预定尺度对应的第二原始中间向量。对与第1个第二预定尺度对应的第二原始中间向量和与第2个第二预定尺度对应的第二目标中间向量进行点积操作,得到与第1个第二预定尺度对应的第二目标中间向量。对与第1个第二预定尺度对应的第二目标中间向量进行卷积操作,得到查询图像的图像分割结果304。其他部分可以参见与图3C对应的实施例相关部分内容的说明,在此不再赘述。
图4示意性示出了根据本公开实施例的图像分割模型的训练方法的流程图。
如图4所示,该方法400包括操作S410~S440。
在操作S410,根据样本查询图像的样本查询图像数据、样本支持图像的样本支持图像数据和与样本支持图像对应的样本掩码图像的样本掩码图像数据,得到至少一个尺度的样本特征信息。样本特征信息包括样本查询图像的样本查询特征信息、样本支持图像的样本支持特征信息和样本掩码图像的样本掩码特征信息。
在操作S420,根据至少一个尺度的样本特征信息,得到至少一个第一预定尺度的样本相关信息。样本相关信息包括以下至少一项:样本查询图像和样本支持图像之间的互相关样本语义信息以及样本查询图像和样本支持图像之间的互关联样本信息。样本查询图像和样本支持图像之间的互相关样本语义信息是根据样本支持图像的第一自相关样本语义信息和样本查询图像的第二自相关样本语义信息确定的。样本查询图像和样本支持图像之间的互关联样本信息是根据样本支持图像的第一自关联样本信息和样本查询图像的第二自关联样本信息确定的。
在操作S430,根据至少一个第二预定尺度的样本特征信息和,至少一个第一预定尺度的样本相关信息,得到样本查询图像的预测图像分割结果。
在操作S440,根据样本查询图像的真实图像分割结果和预测图像分割结果训练图像分割模型,得到训练完成的图像分割模型。
根据本公开的实施例,样本查询图像可以指需要进行图像分割的图像。样本支持图像可以对样本查询图像的图像分割结果起到引导作用。样本查询图像可以包括至少一个对象。每个对象具有与该对象对应的类别。由此,样本查询图像的类别可以包括至少一个。针对样本查询图像包括的每个类别,可以具有与该类别对应的至少一个样本支持图像。与该类别对应的样本支持图像包括与该类别对应的对象。即,可以根据样本查询图像包括的类别,确定与样本查询图像对应的样本支持图像。样本支持图像的数目可以包括多个。具有与样本支持图像对应的样本掩码图像。样本掩码图像可以指样本支持图像的真实图像分割结果。
根据本公开的实施例,图像分割模型可以包括特征提取模块、至少一个关系引导的上下文转换器和至少一个关系引导的关联转换器和解码器。特征提取模块可以包括编码器。针对特征提取模块、至少一个关系引导的上下文转换器和至少一个关系引导的关联转换器和解码器的说明,可以参见上文相关部分内容,在此不再赘述。
根据本公开的实施例,可以基于损失函数,利用真实图像分割结果和预测图像分割结果,得到输出值,根据输出值调节图像分割模型的模型参数,直至满足预定条件。将在满足预定条件的情况下得到的图像分割模型确定为训练完成的图像分割模型。预定条件可以包括以下至少一项:输出值收敛和训练轮次达到最大训练轮次。
根据本公开的实施例,在样本相关信息包括样本查询图像和样本支持图像之间的互相关样本语义信息的情况下,操作S420可以包括如下操作。
针对至少一个第一预定尺度中的每个第一预定尺度,根据与第一预定尺度对应的样本支持特征信息和样本掩码特征信息,得到与第一预定尺度对应的第一自相关样本语义信息。根据与第一预定尺度对应的样本查询特征信息,得到与第一预定尺度对应的第二自相关样本语义信息。根据与第一预定尺度对应的样本支持特征信息、第一自相关样本语义信息和第二自相关样本语义信息,得到与第一预定尺度对应的互相关样本语义信息。
根据本公开的实施例,根据与第一预定尺度对应的样本支持特征信息和样本掩码特征信息,得到与第一预定尺度对应的第一自相关样本语义信息,可以包括如下操作。
基于第一自注意力策略,根据与第一预定尺度对应的样本支持特征信息和样本掩码特征信息,得到与第一预定尺度对应的第一中间样本信息。根据与第一预定尺度对应的第一中间样本信息和样本掩码特征信息,得到与第一预定尺度对应的第一自相关样本语义信息。
根据本公开的实施例,根据与第一预定尺度对应的样本查询特征信息,得到与第一预定尺度对应的第二自相关样本语义信息,可以包括如下操作。
基于第二自注意力策略,根据与第一预定尺度对应的样本查询特征信息,得到与第一预定尺度对应的第二中间样本信息。根据与第一预定尺度对应的第二中间样本信息和样本查询特征信息,得到与第一预定尺度对应的第二自相关样本语义信息。
根据本公开的实施例,根据与第一预定尺度对应的样本支持特征信息、第一自相关样本语义信息和第二自相关样本语义信息,得到与第一预定尺度对应的互相关样本语义信息,可以包括如下操作。
基于第三自注意力策略,根据与第一预定尺度对应的样本支持特征信息、第一自相关样本语义信息和第二自相关样本语义信息,得到与第一预定尺度对应的第三中间样本信息。根据与第一预定尺度对应的第一自相关语义信息和第三中间样本信息,得到与第一预定尺度对应的互相关样本语义信息。
根据本公开的实施例,在样本相关信息包括样本查询图像和样本支持图像之间的互关联样本信息的情况下,操作S420可以包括如下操作。
针对至少一个第一预定尺度中的每个第一预定尺度,根据与第一预定尺度对应的样本支持特征信息和样本掩码特征信息,得到与第一预定尺度对应的第一自关联样本信息。根据与第一预定尺度对应的样本查询特征信息,得到与第一预定尺度对应的第二自关联样本信息。根据与第一预定尺度对应的样本支持特征信息和样本查询特征信息,得到与第一预定尺度对的中间互关联样本信息。根据与第一预定尺度对应的第一自关联样本信息、第二自关联样本信息和中间互关联样本信息,得到与第一预定尺度对应的互关联样本信息。
根据本公开的实施例,根据与第一预定尺度对应的样本支持特征信息和样本掩码特征信息,得到与第一预定尺度对应的第一自关联样本信息,可以包括如下操作。
对与第一预定尺度对应的样本支持特征信息和样本掩码特征信息进行点积操作,得到与第一预定尺度对应的第四中间样本信息。对与第一预定尺度对应的第四中间样本信息进行全连接操作,得到与第一预定尺度对应的第一自关联样本信息。
根据本公开的实施例,根据与第一预定尺度对应的样本查询特征信息,得到与第一预定尺度对应的第二自关联样本信息,可以包括如下操作。
对与第一预定尺度对应的样本查询特征信息进行全连接操作,得到与第一预定尺度对应的第二自关联样本信息。
根据本公开的实施例,根据与第一预定尺度对应的样本支持特征信息和样本查询特征信息,得到与第一预定尺度对的中间互关联样本信息,可以包括如下操作。
对与第一预定尺度对应的样本支持特征信息和样本查询特征信息进行全连接操作,得到与第一预定尺度对应的中间互关联样本信息。
根据本公开的实施例,根据与第一预定尺度对应的第一自关联样本信息、第二自关联样本信息和中间互关联样本信息,得到与第一预定尺度对应的互关联样本信息,可以包括如下操作。
基于第四自注意力策略,根据与第一预定尺度对应的第一自关联样本信息和中间互关联样本信息,得到与第一预定尺度对应的第五中间样本信息。根据与第一预定尺度对应的第二自关联样本信息和第五中间样本信息,得到与第一预定尺度对应的互关联样本信息。
根据本公开的实施例,操作S430可以包括如下操作。
根据至少一个第一预定尺度的样本相关信息,得到第六中间样本信息。根据第六中间样本信息和至少一个第二预定尺度的样本查询特征信息,得到样本查询图像的预测图像分割结果。
根据本公开的实施例,至少一个尺度包括N个尺度。至少一个第一预定尺度包括M个第一预定尺度。至少一个第二预定尺度包括(N-M)个第二预定尺度。N是大于1的整数。M是大于或等于1且小于N的整数。
根据本公开的实施例,在样本相关信息包括样本查询图像和样本支持图像之间的互相关样本语义信息以及样本查询图像和样本支持图像之间的互关联样本信息中的一项的情况下,根据至少一个第一预定尺度的样本相关信息,得到第六中间样本信息,可以包括如下操作。
在(N-M+1)≤i<N的情况下,对与第i个第一预定尺度对应的样本相关信息进行卷积操作,得到与第i个第一预定尺度对应的第一原始中间样本信息。对与第i个第一预定尺度对应的第一原始中间样本信息和与第(i+1)个第一预定尺度对应的第一目标中间样本信息进行点积操作,得到与第i个第一预定尺度对应的第一目标中间样本信息。将与第(N-M+1)个第一预定尺度对应的第一目标中间样本信息确定为第六中间样本信息。
根据本公开的实施例,根据第六中间样本信息和至少一个第二预定尺度的样本查询特征信息,得到样本查询图像的预测图像分割结果,可以包括如下操作。
在j=N-M的情况下,对与第(N-M)个第二预定尺度对应的样本查询特征信息进行卷积操作,得到与第(N-M)个第二预定尺度对应的第一原始中间样本信息。对第六中间样本信息和与第(N-M)个第二预定尺度对应的第一原始中间样本信息进行点积操作,得到与第(N-M)个第二预定尺度对应的第一目标中间样本信息。在1≤j<N-M的情况下,对与第j个第二预定尺度对应的样本查询特征信息进行卷积操作,得到与第j个第二预定尺度对应的第一原始中间样本信息。对与第j个第二预定尺度对应的第一第一原始中间样本信息和与第(j+1)个第二预定尺度对应的第一目标中间样本信息进行点积操作,得到与第j个第二预定尺度对应的第一目标中间样本信息。根据与第1个第二预定尺度对应的第一目标中间样本信息,确定样本查询图像的预测图像分割结果。
根据本公开的实施例,至少一个尺度包括P个尺度。至少一个第一预定尺度包括Q个第一预定尺度。至少一个第二预定尺度包括(P-Q)个第二预定尺度。P是大于1的整数。Q是大于或等于1且小于P的整数。
根据本公开的实施例,在样本相关信息包括样本查询图像和样本支持图像之间的互相关样本语义信息以及样本查询图像和样本支持图像之间的互关联样本信息的情况下,根据至少一个第一预定尺度的样本相关信息,得到第六中间样本信息,可以包括如下操作。
在(P-Q+1)≤k<P的情况下,对与第k个第一预定尺度对应的互相关样本语义信息和互关联样本信息进行点积操作,得到与第k个第一预定尺度对应的第二原始中间样本信息。对与第k个第一预定尺度对应的第二原始中间样本信息进行卷积操作,得到与第k个第一预定尺度对应的第三原始中间样本信息。对与第k个第一预定尺度对应的第三原始中间样本信息和与第(k+1)个第一预定尺度对应的第二目标中间样本信息进行点积操作,得到与第k个第一预定尺度对应的第二目标中间样本信息。将与第(P-Q+1)个第一预定尺度对应的第二目标中间样本信息确定为第六中间信息。
根据本公开的实施例,根据第六中间样本信息和至少一个第二预定尺度的样本查询特征信息,得到样本查询图像的预测图像分割结果,可以包括如下操作。
在l=P-Q的情况下,对与第(P-Q)个第二预定尺度对应的样本查询特征信息进行卷积操作,得到与第(P-Q)个第二预定尺度对应的第二原始中间样本信息。对第六中间样本信息和与第(P-Q)个第二预定尺度对应的第二原始中间样本信息进行点积操作,得到与第(P-Q)个第二预定尺度对应的第二目标中间样本信息。在1≤l<P-Q的情况下,对与第l个第二预定尺度对应的样本查询特征信息进行卷积操作,得到与第l个第二预定尺度对应的第二原始中间样本信息。对与第l个第二预定尺度对应的第二原始中间样本信息和与第(l+1)个第二预定尺度对应的第二目标中间样本信息进行点积操作,得到与第l个第二预定尺度对应的第二目标中间样本信息。根据与第1个第二预定尺度对应的第二目标中间样本信息,确定样本查询图像的预测图像分割结果。
本公开的技术方案中,所涉及的用户个人信息的收集、存储、使用、加工、传输、提供和公开等处理,均符合相关法律法规的规定,且不违背公序良俗。
以上仅是示例性实施例,但不限于此,还可以包括本领域已知的其他图像分割方法和图像分割模型的训练方法,只要能够提高图像分割精度即可。
图5示意性示出了根据本公开实施例的图像分割装置的框图。
如图5所示,图像分割装置500可以包括第一获得模块510、第二获得模块520和第三获得模块530。
第一获得模块510,用于根据查询图像的查询图像数据、支持图像的支持图像数据和与支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息。特征信息包括查询图像的查询特征信息、支持图像的支持特征信息和掩码图像的掩码特征信息。
第二获得模块520,用于根据至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息。相关信息包括以下至少一项:查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息,互相关语义信息是根据支持图像的第一自相关语义信息和查询图像的第二自相关语义信息确定的,互关联信息是根据支持图像的第一自关联信息和查询图像的第二自关联信息确定的。
第三获得模块530,用于根据至少一个第二预定尺度的特征信息和至少一个第一预定尺度的相关信息,得到查询图像的图像分割结果。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息的情况下,第二获得模块520可以包括第一获得子模块、第二获得子模块和第三获得子模块。
针对至少一个第一预定尺度中的每个第一预定尺度,
第一获得子模块,用于根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一自相关语义信息。
第二获得子模块,用于根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二自相关语义信息。
第三获得子模块,用于根据与第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与第一预定尺度对应的互相关语义信息。
根据本公开的实施例,第一获得子模块可以包括第一获得单元和第二获得单元。
第一获得单元,用于基于第一自注意力策略,根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一中间信息。
第二获得单元,用于根据与第一预定尺度对应的第一中间信息和掩码特征信息,得到与第一预定尺度对应的第一自相关语义信息。
根据本公开的实施例,第二获得子模块可以包括第三获得单元和第四获得单元。
第三获得单元,用于基于第二自注意力策略,根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二中间信息。
第四获得单元,用于根据与第一预定尺度对应的第二中间信息和查询特征信息,得到与第一预定尺度对应的第二自相关语义信息。
根据本公开的实施例,第三获得子模块可以包括第五获得单元和第六获得单元。
第五获得单元,用于基于第三自注意力策略,根据与第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与第一预定尺度对应的第三中间信息。
第六获得单元,用于根据与第一预定尺度对应的第一自相关语义信息和第三中间信息,得到与第一预定尺度对应的互相关语义信息。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互关联信息的情况下,第二获得模块520可以包括第四获得子模块、第五获得子模块、第六获得子模块和第七获得子模块。
针对至少一个第一预定尺度中的每个第一预定尺度,
第四获得子模块,用于根据与第一预定尺度对应的支持特征信息和掩码特征信息,得到与第一预定尺度对应的第一自关联信息。
第五获得子模块,用于根据与第一预定尺度对应的查询特征信息,得到与第一预定尺度对应的第二自关联信息。
第六获得子模块,用于根据与第一预定尺度对应的支持特征信息和查询特征信息,得到与第一预定尺度对的中间互关联信息。
第七获得子模块,用于根据与第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息,得到与第一预定尺度对应的互关联信息。
根据本公开的实施例,第四获得子模块可以包括第七获得单元和第八获得单元。
第七获得单元,用于对与第一预定尺度对应的支持特征信息和掩码特征信息进行点积操作,得到与第一预定尺度对应的第四中间信息。
第八获得单元,用于对与第一预定尺度对应的第四中间信息进行全连接操作,得到与第一预定尺度对应的第一自关联信息。
根据本公开的实施例,第五获得子模块可以包括第九获得单元。
第九获得单元,用于对与第一预定尺度对应的查询特征信息进行全连接操作,得到与第一预定尺度对应的第二自关联信息。
根据本公开的实施例,第六获得子模块可以包括第十获得单元。
第十获得单元,用于对与第一预定尺度对应的支持特征信息和查询特征信息进行全连接操作,得到与第一预定尺度对应的中间互关联信息。
根据本公开的实施例,第七获得子模块可以包括第十一获得单元和第十二获得单元。
第十一获得单元,用于基于第四自注意力策略,根据与第一预定尺度对应的第一自关联信息和中间互关联信息,得到与第一预定尺度对应的第五中间信息。
第十二获得单元,用于根据与第一预定尺度对应的第二自关联信息和第五中间信息,得到与第一预定尺度对应的互关联信息。
根据本公开的实施例,第三获得模块530可以包括第八获得子模块和第九获得子模块。
第八获得子模块,用于根据至少一个第一预定尺度的相关信息,得到第六中间信息。
第九获得子模块,用于根据至少一个第二预定尺度的查询特征信息和第六中间信息,得到查询图像的图像分割结果。
根据本公开的实施例,至少一个尺度包括N个尺度。至少一个第一预定尺度包括M个第一预定尺度。至少一个第二预定尺度包括(N-M)个第二预定尺度。N是大于1的整数。M是大于或等于1且小于N的整数。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息中的一项的情况下,第八获得子模块可以包括第十三获得单元、第十四获得单元和第一确定单元。
在(N-M+1)≤i<N的情况下,
第十三获得单元,用于对与第i个第一预定尺度对应的相关信息进行卷积操作,得到与第i个第一预定尺度对应的第一原始中间信息。
第十四获得单元,用于对与第i个第一预定尺度对应的第一原始中间信息和与第(i+1)个第一预定尺度对应的第一目标中间信息进行点积操作,得到与第i个第一预定尺度对应的第一目标中间信息。
第一确定单元,用于将与第(N-M+1)个第一预定尺度对应的第一目标中间信息确定为第六中间信息。
根据本公开的实施例,第九获得子模块可以包括第十五获得单元、第十六获得单元、第十七获得单元、第十八获得单元和第二确定单元。
在j=N-M的情况下,
第十五获得单元,用于对与第(N-M)个第二预定尺度对应的查询特征信息进行卷积操作,得到与第(N-M)个第二预定尺度对应的第一原始中间信息。
第十六获得单元,用于对与第(N-M)个第二预定尺度对应的第一原始中间信息和第六中间信息进行点积操作,得到与第(N-M)个第二预定尺度对应的第一目标中间信息。
在1≤j<N-M的情况下,
第十七获得单元,用于对与第j个第二预定尺度对应的查询特征信息进行卷积操作,得到与第j个第二预定尺度对应的第一原始中间信息。
第十八获得单元,用于对与第j个第二预定尺度对应的第一第一原始中间信息和与第(j+1)个第二预定尺度对应的第一目标中间信息进行点积操作,得到与第j个第二预定尺度对应的第一目标中间信息。
第二确定单元,用于根据与第1个第二预定尺度对应的第一目标中间信息,确定查询图像的图像分割结果。
根据本公开的实施例,至少一个尺度包括P个尺度。至少一个第一预定尺度包括Q个第一预定尺度。至少一个第二预定尺度包括(P-Q)个第二预定尺度。P是大于1的整数。Q是大于或等于1且小于P的整数。
根据本公开的实施例,在相关信息包括查询图像和支持图像之间的互相关语义信息以及查询图像和支持图像之间的互关联信息的情况下,第八获得子模块可以包括第十九获得单元、第二十获得单元、第二十一获得单元和第三确定单元。
在(P-Q+1)≤k<P的情况下,
第十九获得单元,用于对与第k个第一预定尺度对应的互相关语义信息和互关联信息进行点积操作,得到与第k个第一预定尺度对应的第二原始中间信息。
第二十获得单元,用于对与第k个第一预定尺度对应的第二原始中间信息进行卷积操作,得到与第k个第一预定尺度对应的第三原始中间信息。
第二十一获得单元,用于对与第k个第一预定尺度对应的第三原始中间信息和与第(k+1)个第一预定尺度对应的第二目标中间信息进行点积操作,得到与第k个第一预定尺度对应的第二目标中间信息。
第三确定单元,用于将与第(P-Q+1)个第一预定尺度对应的第二目标中间信息确定为第六中间信息。
根据本公开的实施例,第九获得子模块可以包括第二十二获得单元、第二十三获得单元、第二十四获得单元、第二十五获得单元和第四确定单元。
在l=P-Q的情况下,
第二十二获得单元,用于对与第(P-Q)个第二预定尺度对应的查询特征信息进行卷积操作,得到与第(P-Q)个第二预定尺度对应的第二原始中间信息。
第二十三获得单元,用于对与第(P-Q)个第二预定尺度对应的第二原始中间信息和第六中间信息进行点积操作,得到与第(P-Q)个第二预定尺度对应的第二目标中间信息。
在1≤l<P-Q的情况下,
第二十四获得单元,用于对与第l个第二预定尺度对应的查询特征信息进行卷积操作,得到与第l个第二预定尺度对应的第二原始中间信息。
第二十五获得单元,用于对与第l个第二预定尺度对应的第二原始中间信息和与第(l+1)个第二预定尺度对应的第二目标中间信息进行点积操作,得到与第l个第二预定尺度对应的第二目标中间信息。
第四确定单元,用于根据与第1个第二预定尺度对应的第二目标中间信息,确定查询图像的图像分割结果。
图6示意性示出了根据本公开实施例的图像分割模型的训练装置的框图。
如图6所示,图像分割模型的训练装置600可以包括第四获得模块610、第五获得模块620、第六获得模块630和第七获得模块640。
第四获得模块610,用于根据样本查询图像的样本查询图像数据、样本支持图像的样本支持图像数据和与所述样本支持图像对应的样本掩码图像的样本掩码图像数据,得到至少一个尺度的样本特征信息。所述样本特征信息包括所述样本查询图像的样本查询特征信息、所述样本支持图像的样本支持特征信息和所述样本掩码图像的样本掩码特征信息。
第五获得模块620,用于根据所述至少一个尺度的样本特征信息,得到至少一个第一预定尺度的样本相关信息。所述样本相关信息包括以下至少一项:所述样本查询图像和所述样本支持图像之间的互相关样本语义信息以及所述样本查询图像和所述样本支持图像之间的互关联样本信息,所述样本查询图像和所述样本支持图像之间的互相关样本语义信息是根据所述样本支持图像的第一自相关样本语义信息和所述样本查询图像的第二自相关样本语义信息确定的,所述样本查询图像和所述样本支持图像之间的互关联样本信息是根据所述样本支持图像的第一自关联样本信息和所述样本查询图像的第二自关联样本信息确定的。
第六获得模块630,用于根据至少一个第二预定尺度的样本特征信息和所述至少一个第一预定尺度的样本相关信息,得到所述样本查询图像的预测图像分割结果。
第七获得模块640,用于根据所述样本查询图像的真实图像分割结果和预测图像分割结果训练所述图像分割模型,得到训练完成的图像分割模型。
根据本公开的实施例,本公开还提供了一种电子设备、一种可读存储介质和一种计算机程序产品。
根据本公开的实施例,一种电子设备,包括:至少一个处理器;以及与至少一个处理器通信连接的存储器;其中,存储器存储有可被至少一个处理器执行的指令,指令被至少一个处理器执行,以使至少一个处理器能够执行如上所述的方法。
根据本公开的实施例,一种存储有计算机指令的非瞬时计算机可读存储介质,其中,计算机指令用于使计算机执行如上所述的方法。
根据本公开的实施例,一种计算机程序产品,包括计算机程序,计算机程序在被处理器执行时实现如上所述的方法。
图7示意性示出了根据本公开实施例的适于实现图像分割方法和图像分割模型的训练方法的电子设备的框图。电子设备旨在表示各种形式的数字计算机,诸如,膝上型计算机、台式计算机、工作台、个人数字助理、服务器、刀片式服务器、大型计算机、和其它适合的计算机。电子设备还可以表示各种形式的移动装置,诸如,个人数字处理、蜂窝电话、智能电话、可穿戴设备和其它类似的计算装置。本文所示的部件、它们的连接和关系、以及它们的功能仅仅作为示例,并且不意在限制本文中描述的和/或者要求的本公开的实现。
如图7所示,电子设备700包括计算单元701,其可以根据存储在只读存储器(ROM)702中的计算机程序或者从存储单元708加载到随机访问存储器(RAM)703中的计算机程序,来执行各种适当的动作和处理。在RAM 703中,还可存储电子设备700操作所需的各种程序和数据。计算单元701、ROM 702以及RAM 703通过总线704彼此相连。输入/输出(I/O)接口705也连接至总线704。
电子设备700中的多个部件连接至I/O接口705,包括:输入单元706,例如键盘、鼠标等;输出单元707,例如各种类型的显示器、扬声器等;存储单元708,例如磁盘、光盘等;以及通信单元709,例如网卡、调制解调器、无线通信收发机等。通信单元709允许电子设备700通过诸如因特网的计算机网络和/或各种电信网络与其他设备交换信息/数据。
计算单元701可以是各种具有处理和计算能力的通用和/或专用处理组件。计算单元701的一些示例包括但不限于中央处理单元(CPU)、图形处理单元(GPU)、各种专用的人工智能(AI)计算芯片、各种运行机器学习模型算法的计算单元、数字信号处理器(DSP)、以及任何适当的处理器、控制器、微控制器等。计算单元701执行上文所描述的各个方法和处理,例如图像分割方法或图像分割模型的训练方法。例如,在一些实施例中,图像分割方法或图像分割模型的训练方法可被实现为计算机软件程序,其被有形地包含于机器可读介质,例如存储单元708。在一些实施例中,计算机程序的部分或者全部可以经由ROM 702和/或通信单元709而被载入和/或安装到电子设备700上。当计算机程序加载到RAM 703并由计算单元701执行时,可以执行上文描述的图像分割方法或图像分割模型的训练方法的一个或多个步骤。备选地,在其他实施例中,计算单元701可以通过其他任何适当的方式(例如,借助于固件)而被配置为执行图像分割方法或图像分割模型的训练方法。
本文中以上描述的系统和技术的各种实施方式可以在数字电子电路系统、集成电路系统、场可编程门阵列(FPGA)、专用集成电路(ASIC)、专用标准产品(ASSP)、芯片上系统的系统(SOC)、复杂可编程逻辑设备(CPLD)、计算机硬件、固件、软件、和/或它们的组合中实现。这些各种实施方式可以包括:实施在一个或者多个计算机程序中,该一个或者多个计算机程序可在包括至少一个可编程处理器的可编程系统上执行和/或解释,该可编程处理器可以是专用或者通用可编程处理器,可以从存储系统、至少一个输入装置、和至少一个输出装置接收数据和指令,并且将数据和指令传输至该存储系统、该至少一个输入装置、和该至少一个输出装置。
用于实施本公开的方法的程序代码可以采用一个或多个编程语言的任何组合来编写。这些程序代码可以提供给通用计算机、专用计算机或其他可编程数据处理装置的处理器或控制器,使得程序代码当由处理器或控制器执行时使流程图和/或框图中所规定的功能/操作被实施。程序代码可以完全在机器上执行、部分地在机器上执行,作为独立软件包部分地在机器上执行且部分地在远程机器上执行或完全在远程机器或服务器上执行。
在本公开的上下文中,机器可读介质可以是有形的介质,其可以包含或存储以供指令执行系统、装置或设备使用或与指令执行系统、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体系统、装置或设备,或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦除可编程只读存储器(EPROM或快闪存储器)、光纤、便捷式紧凑盘只读存储器(CD-ROM)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
为了提供与用户的交互,可以在计算机上实施此处描述的系统和技术,该计算机具有:用于向用户显示信息的显示装置(例如,CRT(阴极射线管)或者LCD(液晶显示器)监视器);以及键盘和指向装置(例如,鼠标或者轨迹球),用户可以通过该键盘和该指向装置来将输入提供给计算机。其它种类的装置还可以用于提供与用户的交互;例如,提供给用户的反馈可以是任何形式的传感反馈(例如,视觉反馈、听觉反馈、或者触觉反馈);并且可以用任何形式(包括声输入、语音输入或者、触觉输入)来接收来自用户的输入。
可以将此处描述的系统和技术实施在包括后台部件的计算系统(例如,作为数据服务器)、或者包括中间件部件的计算系统(例如,应用服务器)、或者包括前端部件的计算系统(例如,具有图形用户界面或者网络浏览器的用户计算机,用户可以通过该图形用户界面或者该网络浏览器来与此处描述的系统和技术的实施方式交互)、或者包括这种后台部件、中间件部件、或者前端部件的任何组合的计算系统中。可以通过任何形式或者介质的数字数据通信(例如,通信网络)来将系统的部件相互连接。通信网络的示例包括:局域网(LAN)、广域网(WAN)和互联网。
计算机系统可以包括客户端和服务器。客户端和服务器一般远离彼此并且通常通过通信网络进行交互。通过在相应的计算机上运行并且彼此具有客户端-服务器关系的计算机程序来产生客户端和服务器的关系。服务器可以是云服务器,也可以是分布式系统的服务器,或者是结合了区块链的服务器。
应该理解,可以使用上面所示的各种形式的流程,重新排序、增加或删除步骤。例如,本发公开中记载的各步骤可以并行地执行也可以顺序地执行也可以不同的次序执行,只要能够实现本公开公开的技术方案所期望的结果,本文在此不进行限制。
上述具体实施方式,并不构成对本公开保护范围的限制。本领域技术人员应该明白的是,根据设计要求和其他因素,可以进行各种修改、组合、子组合和替代。任何在本公开的精神和原则之内所作的修改、等同替换和改进等,均应包含在本公开保护范围之内。
Claims (35)
1.一种图像分割方法,包括:
根据查询图像的查询图像数据、支持图像的支持图像数据和与所述支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息,其中,所述特征信息包括所述查询图像的查询特征信息、所述支持图像的支持特征信息和所述掩码图像的掩码特征信息;
根据所述至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,其中,所述相关信息包括以下至少一项:所述查询图像和所述支持图像之间的互相关语义信息以及所述查询图像和所述支持图像之间的互关联信息,所述互相关语义信息是根据所述支持图像的第一自相关语义信息和所述查询图像的第二自相关语义信息确定的,所述互关联信息是根据所述支持图像的第一自关联信息和所述查询图像的第二自关联信息确定的;以及
根据至少一个第二预定尺度的特征信息和所述至少一个第一预定尺度的相关信息,得到所述查询图像的图像分割结果。
2.根据权利要求1所述的方法,其中,在所述相关信息包括所述查询图像和所述支持图像之间的互相关语义信息的情况下,所述根据所述至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,包括:
针对所述至少一个第一预定尺度中的每个第一预定尺度,
根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一自相关语义信息;
根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二自相关语义信息;以及
根据与所述第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与所述第一预定尺度对应的互相关语义信息。
3.根据权利要求2所述的方法,其中,所述根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一自相关语义信息,包括:
基于第一自注意力策略,根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一中间信息;以及
根据与所述第一预定尺度对应的第一中间信息和掩码特征信息,得到与所述第一预定尺度对应的第一自相关语义信息。
4.根据权利要求2或3所述的方法,其中,所述根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二自相关语义信息,包括:
基于第二自注意力策略,根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二中间信息;以及
根据与所述第一预定尺度对应的第二中间信息和查询特征信息,得到与所述第一预定尺度对应的第二自相关语义信息。
5.根据权利要求2~4中任一项所述的方法,其中,所述根据与所述第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与所述第一预定尺度对应的互相关语义信息,包括:
基于第三自注意力策略,根据与所述第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与所述第一预定尺度对应的第三中间信息;以及
根据与所述第一预定尺度对应的第一自相关语义信息和第三中间信息,得到与所述第一预定尺度对应的互相关语义信息。
6.根据权利要求2~5中任一项所述的方法,其中,在所述相关信息包括所述查询图像和所述支持图像之间的互关联信息的情况下,所述根据所述至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,包括:
针对所述至少一个第一预定尺度中的每个第一预定尺度,
根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一自关联信息;
根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二自关联信息;
根据与所述第一预定尺度对应的支持特征信息和查询特征信息,得到与所述第一预定尺度对的中间互关联信息;以及
根据与所述第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息,得到与所述第一预定尺度对应的互关联信息。
7.根据权利要求6所述的方法,其中,所述根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一自关联信息,包括:
对与所述第一预定尺度对应的支持特征信息和掩码特征信息进行点积操作,得到与所述第一预定尺度对应的第四中间信息;以及
对与所述第一预定尺度对应的第四中间信息进行全连接操作,得到与所述第一预定尺度对应的第一自关联信息。
8.根据权利要求6或7所述的方法,其中,所述根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二自关联信息,包括:
对与所述第一预定尺度对应的查询特征信息进行全连接操作,得到与所述第一预定尺度对应的第二自关联信息。
9.根据权利要求6~8中任一项所述的方法,其中,所述根据与所述第一预定尺度对应的支持特征信息和查询特征信息,得到与所述第一预定尺度对的中间互关联信息,包括:
对与所述第一预定尺度对应的支持特征信息和查询特征信息进行全连接操作,得到与所述第一预定尺度对应的中间互关联信息。
10.根据权利要求6~9中任一项所述的方法,其中,所述根据与所述第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息,得到与所述第一预定尺度对应的互关联信息,包括:
基于第四自注意力策略,根据与所述第一预定尺度对应的第一自关联信息和中间互关联信息,得到与所述第一预定尺度对应的第五中间信息;以及
根据与所述第一预定尺度对应的第二自关联信息和第五中间信息,得到与所述第一预定尺度对应的互关联信息。
11.根据权利要求1~10中任一项所述的方法,其中,所述根据至少一个第二预定尺度的特征信息和所述至少一个第一预定尺度的相关信息,得到所述查询图像的图像分割结果,包括:
根据所述至少一个第一预定尺度的相关信息,得到第六中间信息;以及
根据所述至少一个第二预定尺度的查询特征信息和所述第六中间信息,得到所述查询图像的图像分割结果。
12.根据权利要求11所述的方法,其中,所述至少一个尺度包括N个尺度,所述至少一个第一预定尺度包括M个第一预定尺度,所述至少一个第二预定尺度包括(N-M)个第二预定尺度,N是大于1的整数,M是大于或等于1且小于N的整数;
其中,在所述相关信息包括所述查询图像和所述支持图像之间的互相关语义信息以及所述查询图像和所述支持图像之间的互关联信息中的一项的情况下,所述根据所述至少一个第一预定尺度的相关信息,得到第六中间信息,包括:
在(N-M+1)≤i<N的情况下,
对与第i个第一预定尺度对应的相关信息进行卷积操作,得到与所述第7个第一预定尺度对应的第一原始中间信息;以及
对与所述第i个第一预定尺度对应的第一原始中间信息和与第(i+1)个第一预定尺度对应的第一目标中间信息进行点积操作,得到与所述第i个第一预定尺度对应的第一目标中间信息;
将与第(N-M+1)个第一预定尺度对应的第一目标中间信息确定为所述第六中间信息。
13.根据权利要求12所述的方法,其中,所述根据所述至少一个第二预定尺度的查询特征信息和所述第六中间信息,得到所述查询图像的图像分割结果,包括:
在j=N-M的情况下,
对与第(N-M)个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第(N-M)个第二预定尺度对应的第一原始中间信息;以及
对与所述第(N-M)个第二预定尺度对应的第一原始中间信息和所述第六中间信息进行点积操作,得到与所述第(N-M)个第二预定尺度对应的第一目标中间信息;
在1≤j<N-M的情况下,
对与第j个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第j个第二预定尺度对应的第一原始中间信息;以及
对与所述第j个第二预定尺度对应的第一第一原始中间信息和与第(j+1)个第二预定尺度对应的第一目标中间信息进行点积操作,得到与所述第j个第二预定尺度对应的第一目标中间信息;
根据与第1个第二预定尺度对应的第一目标中间信息,确定所述查询图像的图像分割结果。
14.根据权利要求11所述的方法,其中,所述至少一个尺度包括P个尺度,所述至少一个第一预定尺度包括Q个第一预定尺度,所述至少一个第二预定尺度包括(P-Q)个第二预定尺度,P是大于1的整数,Q是大于或等于1且小于P的整数;
其中,在所述相关信息包括所述查询图像和所述支持图像之间的互相关语义信息以及所述查询图像和所述支持图像之间的互关联信息的情况下,所述根据所述至少一个第一预定尺度的相关信息,得到第六中间信息,包括:
在(P-Q+1)≤k<P的情况下,
对与第k个第一预定尺度对应的互相关语义信息和互关联信息进行点积操作,得到与所述第k个第一预定尺度对应的第二原始中间信息;
对与所述第k个第一预定尺度对应的第二原始中间信息进行卷积操作,得到与所述第k个第一预定尺度对应的第三原始中间信息;以及
对与所述第k个第一预定尺度对应的第三原始中间信息和与第(k+1)个第一预定尺度对应的第二目标中间信息进行点积操作,得到与所述第k个第一预定尺度对应的第二目标中间信息;
将与第(P-Q+1)个第一预定尺度对应的第二目标中间信息确定为所述第六中间信息。
15.根据权利要求14所述的方法,其中,所述根据所述至少一个第二预定尺度的查询特征信息和所述第六中间信息,得到所述查询图像的图像分割结果,包括:
在l=P-Q的情况下,
对与第(P-Q)个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第(P-Q)个第二预定尺度对应的第二原始中间信息;以及
对与所述第(P-Q)个第二预定尺度对应的第二原始中间信息和所述第六中间信息进行点积操作,得到与所述第(P-Q)个第二预定尺度对应的第二目标中间信息;
在1≤l<P-Q的情况下,
对与第l个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第l个第二预定尺度对应的第二原始中间信息;以及
对与所述第l个第二预定尺度对应的第二原始中间信息和与第(l+1)个第二预定尺度对应的第二目标中间信息进行点积操作,得到与所述第l个第二预定尺度对应的第二目标中间信息;
根据与第1个第二预定尺度对应的第二目标中间信息,确定所述查询图像的图像分割结果。
16.一种图像分割模型的训练方法,包括:
根据样本查询图像的样本查询图像数据、样本支持图像的样本支持图像数据和与所述样本支持图像对应的样本掩码图像的样本掩码图像数据,得到至少一个尺度的样本特征信息,其中,所述样本特征信息包括所述样本查询图像的样本查询特征信息、所述样本支持图像的样本支持特征信息和所述样本掩码图像的样本掩码特征信息;
根据所述至少一个尺度的样本特征信息,得到至少一个第一预定尺度的样本相关信息,其中,所述样本相关信息包括以下至少一项:所述样本查询图像和所述样本支持图像之间的互相关样本语义信息以及所述样本查询图像和所述样本支持图像之间的互关联样本信息,所述样本查询图像和所述样本支持图像之间的互相关样本语义信息是根据所述样本支持图像的第一自相关样本语义信息和所述样本查询图像的第二自相关样本语义信息确定的,所述样本查询图像和所述样本支持图像之间的互关联样本信息是根据所述样本支持图像的第一自关联样本信息和所述样本查询图像的第二自关联样本信息确定的;
根据至少一个第二预定尺度的样本特征信息和所述至少一个第一预定尺度的样本相关信息,得到所述样本查询图像的预测图像分割结果;以及
根据所述样本查询图像的真实图像分割结果和预测图像分割结果训练所述图像分割模型,得到训练完成的图像分割模型。
17.一种图像分割装置,包括:
第一获得模块,用于根据查询图像的查询图像数据、支持图像的支持图像数据和与所述支持图像对应的掩码图像的掩码图像数据,得到至少一个尺度的特征信息,其中,所述特征信息包括所述查询图像的查询特征信息、所述支持图像的支持特征信息和所述掩码图像的掩码特征信息;
第二获得模块,用于根据所述至少一个尺度的特征信息,得到至少一个第一预定尺度的相关信息,其中,所述相关信息包括以下至少一项:所述查询图像和所述支持图像之间的互相关语义信息以及所述查询图像和所述支持图像之间的互关联信息,所述互相关语义信息是根据所述支持图像的第一自相关语义信息和所述查询图像的第二自相关语义信息确定的,所述互关联信息是根据所述支持图像的第一自关联信息和所述查询图像的第二自关联信息确定的;以及
第三获得模块,用于根据至少一个第二预定尺度的特征信息和所述至少一个第一预定尺度的相关信息,得到所述查询图像的图像分割结果。
18.根据权利要求17所述的装置,其中,在所述相关信息包括所述查询图像和所述支持图像之间的互相关语义信息的情况下,所述第二获得模块,包括:
针对所述至少一个第一预定尺度中的每个第一预定尺度,
第一获得子模块,用于根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一自相关语义信息;
第二获得子模块,用于根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二自相关语义信息;以及
第三获得子模块,用于根据与所述第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与所述第一预定尺度对应的互相关语义信息。
19.根据权利要求18所述的装置,其中,所述第一获得子模块,包括:
第一获得单元,用于基于第一自注意力策略,根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一中间信息;以及
第二获得单元,用于根据与所述第一预定尺度对应的第一中间信息和掩码特征信息,得到与所述第一预定尺度对应的第一自相关语义信息。
20.根据权利要求18或19所述的装置,其中,所述第二获得子模块,包括:
第三获得单元,用于基于第二自注意力策略,根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二中间信息;以及
第四获得单元,用于根据与所述第一预定尺度对应的第二中间信息和查询特征信息,得到与所述第一预定尺度对应的第二自相关语义信息。
21.根据权利要求18~20中任一项所述的装置,其中,所述第三获得子模块,包括:
第五获得单元,用于基于第三自注意力策略,根据与所述第一预定尺度对应的支持特征信息、第一自相关语义信息和第二自相关语义信息,得到与所述第一预定尺度对应的第三中间信息;以及
第六获得单元,用于根据与所述第一预定尺度对应的第一自相关语义信息和第三中间信息,得到与所述第一预定尺度对应的互相关语义信息。
22.根据权利要求18~21中任一项所述的装置,其中,在所述相关信息包括所述查询图像和所述支持图像之间的互关联信息的情况下,所述第二获得模块,包括:
针对所述至少一个第一预定尺度中的每个第一预定尺度,
第四获得子模块,用于根据与所述第一预定尺度对应的支持特征信息和掩码特征信息,得到与所述第一预定尺度对应的第一自关联信息;
第五获得子模块,用于根据与所述第一预定尺度对应的查询特征信息,得到与所述第一预定尺度对应的第二自关联信息;
第六获得子模块,用于根据与所述第一预定尺度对应的支持特征信息和查询特征信息,得到与所述第一预定尺度对的中间互关联信息;以及
第七获得子模块,用于根据与所述第一预定尺度对应的第一自关联信息、第二自关联信息和中间互关联信息,得到与所述第一预定尺度对应的互关联信息。
23.根据权利要求22所述的装置,其中,所述第四获得子模块,包括:
第七获得单元,用于对与所述第一预定尺度对应的支持特征信息和掩码特征信息进行点积操作,得到与所述第一预定尺度对应的第四中间信息;以及
第八获得单元,用于对与所述第一预定尺度对应的第四中间信息进行全连接操作,得到与所述第一预定尺度对应的第一自关联信息。
24.根据权利要求22或23所述的装置,其中,所述第五获得子模块,包括:
第九获得单元,用于对与所述第一预定尺度对应的查询特征信息进行全连接操作,得到与所述第一预定尺度对应的第二自关联信息。
25.根据权利要求22~24中任一项所述的装置,其中,所述第六获得子模块,包括:
第十获得单元,用于对与所述第一预定尺度对应的支持特征信息和查询特征信息进行全连接操作,得到与所述第一预定尺度对应的中间互关联信息。
26.根据权利要求22~25中任一项所述的装置,其中,所述第七获得子模块,包括:
第十一获得单元,用于基于第四自注意力策略,根据与所述第一预定尺度对应的第一自关联信息和中间互关联信息,得到与所述第一预定尺度对应的第五中间信息;以及
第十二获得单元,用于根据与所述第一预定尺度对应的第二自关联信息和第五中间信息,得到与所述第一预定尺度对应的互关联信息。
27.根据权利要求17~26中任一项所述的装置,其中,所述第三获得模块,包括:
第八获得子模块,用于根据所述至少一个第一预定尺度的相关信息,得到第六中间信息;以及
第九获得子模块,用于根据所述至少一个第二预定尺度的查询特征信息和所述第六中间信息,得到所述查询图像的图像分割结果。
28.根据权利要求27所述的装置,其中,所述至少一个尺度包括N个尺度,所述至少一个第一预定尺度包括M个第一预定尺度,所述至少一个第二预定尺度包括(N-M)个第二预定尺度,N是大于1的整数,M是大于或等于1且小于N的整数;
其中,在所述相关信息包括所述查询图像和所述支持图像之间的互相关语义信息以及所述查询图像和所述支持图像之间的互关联信息中的一项的情况下,所述第八获得子模块,包括:
在(N-M+1)≤i<N的情况下,
第十三获得单元,用于对与第i个第一预定尺度对应的相关信息进行卷积操作,得到与所述第i个第一预定尺度对应的第一原始中间信息;以及
第十四获得单元,用于对与所述第i个第一预定尺度对应的第一原始中间信息和与第(i+1)个第一预定尺度对应的第一目标中间信息进行点积操作,得到与所述第i个第一预定尺度对应的第一目标中间信息;
第一确定单元,用于将与第(N-M+1)个第一预定尺度对应的第一目标中间信息确定为所述第六中间信息。
29.根据权利要求28所述的装置,其中,所述第九获得子模块,包括:
在j=N-M的情况下,
第十五获得单元,用于对与第(N-M)个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第(N-M)个第二预定尺度对应的第一原始中间信息;以及
第十六获得单元,用于对与所述第(N-M)个第二预定尺度对应的第一原始中间信息和所述第六中间信息进行点积操作,得到与所述第(N-M)个第二预定尺度对应的第一目标中间信息;
在1≤j<N-M的情况下,
第十七获得单元,用于对与第j个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第j个第二预定尺度对应的第一原始中间信息;以及
第十八获得单元,用于对与所述第j个第二预定尺度对应的第一第一原始中间信息和与第(j+1)个第二预定尺度对应的第一目标中间信息进行点积操作,得到与所述第j个第二预定尺度对应的第一目标中间信息;
第二确定单元,用于根据与第1个第二预定尺度对应的第一目标中间信息,确定所述查询图像的图像分割结果。
30.根据权利要求27所述的装置,其中,所述至少一个尺度包括P个尺度,所述至少一个第一预定尺度包括Q个第一预定尺度,所述至少一个第二预定尺度包括(P-Q)个第二预定尺度,P是大于1的整数,Q是大于或等于1且小于P的整数;
其中,在所述相关信息包括所述查询图像和所述支持图像之间的互相关语义信息以及所述查询图像和所述支持图像之间的互关联信息的情况下,所述第八获得子模块,包括:
在(P-Q+1)≤k<P的情况下,
第十九获得单元,用于对与第k个第一预定尺度对应的互相关语义信息和互关联信息进行点积操作,得到与所述第k个第一预定尺度对应的第二原始中间信息;
第二十获得单元,用于对与所述第k个第一预定尺度对应的第二原始中间信息进行卷积操作,得到与所述第k个第一预定尺度对应的第三原始中间信息;以及
第二十一获得单元,用于对与所述第k个第一预定尺度对应的第三原始中间信息和与第(k+1)个第一预定尺度对应的第二目标中间信息进行点积操作,得到与所述第k个第一预定尺度对应的第二目标中间信息;
第三确定单元,用于将与第(P-Q+1)个第一预定尺度对应的第二目标中间信息确定为所述第六中间信息。
31.根据权利要求30所述的装置,其中,所述第九获得子模块,包括:
在l=P-Q的情况下,
第二十二获得单元,用于对与第(P-Q)个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第(P-Q)个第二预定尺度对应的第二原始中间信息;以及
第二十三获得单元,用于对与所述第(P-Q)个第二预定尺度对应的第二原始中间信息和所述第六中间信息进行点积操作,得到与所述第(P-Q)个第二预定尺度对应的第二目标中间信息;
在1≤l<P-Q的情况下,
第二十四获得单元,用于对与第l个第二预定尺度对应的查询特征信息进行卷积操作,得到与所述第l个第二预定尺度对应的第二原始中间信息;以及
第二十五获得单元,用于对与所述第l个第二预定尺度对应的第二原始中间信息和与第(l+1)个第二预定尺度对应的第二目标中间信息进行点积操作,得到与所述第l个第二预定尺度对应的第二目标中间信息;
第四确定单元,用于根据与第1个第二预定尺度对应的第二目标中间信息,确定所述查询图像的图像分割结果。
32.一种图像分割模型的训练装置,包括:
第四获得模块,用于根据样本查询图像的样本查询图像数据、样本支持图像的样本支持图像数据和与所述样本支持图像对应的样本掩码图像的样本掩码图像数据,得到至少一个尺度的样本特征信息,其中,所述样本特征信息包括所述样本查询图像的样本查询特征信息、所述样本支持图像的样本支持特征信息和所述样本掩码图像的样本掩码特征信息;
第五获得模块,用于根据所述至少一个尺度的样本特征信息,得到至少一个第一预定尺度的样本相关信息,其中,所述样本相关信息包括以下至少一项:所述样本查询图像和所述样本支持图像之间的互相关样本语义信息以及所述样本查询图像和所述样本支持图像之间的互关联样本信息,所述样本查询图像和所述样本支持图像之间的互相关样本语义信息是根据所述样本支持图像的第一自相关样本语义信息和所述样本查询图像的第二自相关样本语义信息确定的,所述样本查询图像和所述样本支持图像之间的互关联样本信息是根据所述样本支持图像的第一自关联样本信息和所述样本查询图像的第二自关联样本信息确定的;
第六获得模块,用于根据至少一个第二预定尺度的样本特征信息和所述至少一个第一预定尺度的样本相关信息,得到所述样本查询图像的预测图像分割结果;以及
第七获得模块,用于根据所述样本查询图像的真实图像分割结果和预测图像分割结果训练所述图像分割模型,得到训练完成的图像分割模型。
33.一种电子设备,包括:
至少一个处理器;以及
与所述至少一个处理器通信连接的存储器;其中,
所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1~15中任一项所述的方法或权利要求16所述的方法。
34.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1~15中任一项所述的方法或权利要求16所述的方法。
35.一种计算机程序产品,包括计算机程序,所述计算机程序在被处理器执行时实现根据权利要求1~15中任一项所述的方法或权利要求16所述的方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210174233.8A CN114565760B (zh) | 2022-02-24 | 2022-02-24 | 图像分割方法、模型的训练方法、装置、电子设备及介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210174233.8A CN114565760B (zh) | 2022-02-24 | 2022-02-24 | 图像分割方法、模型的训练方法、装置、电子设备及介质 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114565760A true CN114565760A (zh) | 2022-05-31 |
CN114565760B CN114565760B (zh) | 2023-02-28 |
Family
ID=81715284
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210174233.8A Active CN114565760B (zh) | 2022-02-24 | 2022-02-24 | 图像分割方法、模型的训练方法、装置、电子设备及介质 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114565760B (zh) |
Cited By (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
KR102691077B1 (ko) * | 2023-02-17 | 2024-08-05 | 고려대학교 산학협력단 | 퓨-샷 세그멘테이션 방법 및 그 장치 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200334490A1 (en) * | 2019-04-16 | 2020-10-22 | Fujitsu Limited | Image processing apparatus, training method and training apparatus for the same |
CN112950606A (zh) * | 2021-03-15 | 2021-06-11 | 重庆邮电大学 | 一种基于小样本的手机屏幕缺陷分割方法 |
US20210319263A1 (en) * | 2020-04-13 | 2021-10-14 | International Business Machines Corporation | System and method for augmenting few-shot object classification with semantic information from multiple sources |
CN113569865A (zh) * | 2021-09-27 | 2021-10-29 | 南京码极客科技有限公司 | 一种基于类别原型学习的单样本图像分割方法 |
US20210397876A1 (en) * | 2020-06-19 | 2021-12-23 | Adobe Inc. | Similarity propagation for one-shot and few-shot image segmentation |
-
2022
- 2022-02-24 CN CN202210174233.8A patent/CN114565760B/zh active Active
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200334490A1 (en) * | 2019-04-16 | 2020-10-22 | Fujitsu Limited | Image processing apparatus, training method and training apparatus for the same |
US20210319263A1 (en) * | 2020-04-13 | 2021-10-14 | International Business Machines Corporation | System and method for augmenting few-shot object classification with semantic information from multiple sources |
US20210397876A1 (en) * | 2020-06-19 | 2021-12-23 | Adobe Inc. | Similarity propagation for one-shot and few-shot image segmentation |
CN112950606A (zh) * | 2021-03-15 | 2021-06-11 | 重庆邮电大学 | 一种基于小样本的手机屏幕缺陷分割方法 |
CN113569865A (zh) * | 2021-09-27 | 2021-10-29 | 南京码极客科技有限公司 | 一种基于类别原型学习的单样本图像分割方法 |
Non-Patent Citations (3)
Title |
---|
DAHYUN KANG等: "Relational Embedding for Few-Shot Classification", 《ARXIV:2108.09666V1》 * |
GENGWEI ZHANG等: "Few-Shot Segmentation via Cycle-Consistent Transformer", 《ARXIV:2106.02320V1》 * |
贾熹滨,李佳: "金字塔原型对齐的轻量级小样本语义分割网络", 《北京工业大学学报》 * |
Cited By (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
KR102691077B1 (ko) * | 2023-02-17 | 2024-08-05 | 고려대학교 산학협력단 | 퓨-샷 세그멘테이션 방법 및 그 장치 |
Also Published As
Publication number | Publication date |
---|---|
CN114565760B (zh) | 2023-02-28 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112966522A (zh) | 一种图像分类方法、装置、电子设备及存储介质 | |
CN114612759B (zh) | 视频处理方法、查询视频的方法和模型训练方法、装置 | |
CN114549874B (zh) | 多目标图文匹配模型的训练方法、图文检索方法及装置 | |
CN114020950B (zh) | 图像检索模型的训练方法、装置、设备以及存储介质 | |
CN114429633B (zh) | 文本识别方法、模型的训练方法、装置、电子设备及介质 | |
CN114782719B (zh) | 一种特征提取模型的训练方法、对象检索方法以及装置 | |
US20230186607A1 (en) | Multi-task identification method, training method, electronic device, and storage medium | |
CN115409855B (zh) | 图像处理方法、装置、电子设备和存储介质 | |
CN115456167B (zh) | 轻量级模型训练方法、图像处理方法、装置及电子设备 | |
CN112580733A (zh) | 分类模型的训练方法、装置、设备以及存储介质 | |
CN113887615A (zh) | 图像处理方法、装置、设备和介质 | |
CN114693934A (zh) | 语义分割模型的训练方法、视频语义分割方法及装置 | |
CN114565760B (zh) | 图像分割方法、模型的训练方法、装置、电子设备及介质 | |
CN112906368B (zh) | 行业文本增量方法、相关装置及计算机程序产品 | |
CN114202648A (zh) | 文本图像矫正方法、训练方法、装置、电子设备以及介质 | |
CN116824609B (zh) | 文档版式检测方法、装置和电子设备 | |
CN116311298A (zh) | 信息生成方法、信息处理方法、装置、电子设备以及介质 | |
CN114419327B (zh) | 图像检测方法和图像检测模型的训练方法、装置 | |
CN114707591B (zh) | 数据处理方法和数据处理模型的训练方法、装置 | |
CN113642654B (zh) | 图像特征的融合方法、装置、电子设备和存储介质 | |
CN113139463B (zh) | 用于训练模型的方法、装置、设备、介质和程序产品 | |
CN116229095A (zh) | 一种模型训练方法、视觉任务处理方法、装置及设备 | |
CN113610856B (zh) | 训练图像分割模型和图像分割的方法和装置 | |
CN113361522B (zh) | 用于确定字符序列的方法、装置和电子设备 | |
CN114661904A (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 | ||
GR01 | Patent grant | ||
GR01 | Patent grant |