CN115375656A - 息肉分割模型的训练方法、分割方法、装置、介质及设备 - Google Patents

息肉分割模型的训练方法、分割方法、装置、介质及设备 Download PDF

Info

Publication number
CN115375656A
CN115375656A CN202211015296.5A CN202211015296A CN115375656A CN 115375656 A CN115375656 A CN 115375656A CN 202211015296 A CN202211015296 A CN 202211015296A CN 115375656 A CN115375656 A CN 115375656A
Authority
CN
China
Prior art keywords
domain
target
feature
block
foreground
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
CN202211015296.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.)
Douyin Vision Co Ltd
Original Assignee
Douyin Vision 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 Douyin Vision Co Ltd filed Critical Douyin Vision Co Ltd
Priority to CN202211015296.5A priority Critical patent/CN115375656A/zh
Publication of CN115375656A publication Critical patent/CN115375656A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00—Image analysis
    • G06T7/0002—Inspection of images, e.g. flaw detection
    • G06T7/0012—Biomedical image inspection
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00—Arrangements for image or video recognition or understanding
    • G06V10/20—Image preprocessing
    • G06V10/26—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00—Arrangements for image or video recognition or understanding
    • G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/74—Image or video pattern matching; Proximity measures in feature spaces
    • G06V10/761—Proximity, similarity or dissimilarity measures
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00—Arrangements for image or video recognition or understanding
    • G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/764—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00—Arrangements for image or video recognition or understanding
    • G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
    • G06V10/774—Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00—Arrangements for image or video recognition or understanding
    • G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
    • G06V10/80—Fusion, i.e. combining data from various sources at the sensor level, preprocessing level, feature extraction level or classification level
    • G06V10/806—Fusion, i.e. combining data from various sources at the sensor level, preprocessing level, feature extraction level or classification level of extracted features
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00—Arrangements for image or video recognition or understanding
    • G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00—Indexing scheme for image analysis or image enhancement
    • G06T2207/20—Special algorithmic details
    • G06T2207/20021—Dividing image into blocks, subimages or windows
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00—Indexing scheme for image analysis or image enhancement
    • G06T2207/20—Special algorithmic details
    • G06T2207/20081—Training; Learning
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00—Indexing scheme for image analysis or image enhancement
    • G06T2207/20—Special algorithmic details
    • G06T2207/20084—Artificial neural networks [ANN]

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • General Physics & Mathematics (AREA)
  • Physics & Mathematics (AREA)
  • Medical Informatics (AREA)
  • Health & Medical Sciences (AREA)
  • Evolutionary Computation (AREA)
  • General Health & Medical Sciences (AREA)
  • Multimedia (AREA)
  • Artificial Intelligence (AREA)
  • Computing Systems (AREA)
  • Databases & Information Systems (AREA)
  • Software Systems (AREA)
  • Nuclear Medicine, Radiotherapy & Molecular Imaging (AREA)
  • Radiology & Medical Imaging (AREA)
  • Quality & Reliability (AREA)
  • Image Analysis (AREA)

Abstract

本公开涉及一种息肉分割模型的训练方法、分割方法、装置、介质及设备，所述方法包括：基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征；确定第一源域块特征中的源域前景块特征和第一目标域块特征中的目标域前景块特征；确定源域前景块特征和目标域前景块特征对应的连接关系；根据连接关系对第一块特征进行更新，获得第二块特征；根据第二源域块特征获得源域图像对应的预测分割结果；根据源域图像对应的预测分割结果和息肉分割标签、以及第二源域块特征和第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。

Description

息肉分割模型的训练方法、分割方法、装置、介质及设备
技术领域
本公开涉及图像处理领域，具体地，涉及一种息肉分割模型的训练方法、分割方法、装置、介质及设备。
背景技术
在全球范围内，结直肠癌为经常诊断出的癌症疾病，结肠镜检查可以有效地发现并去除息肉。息肉分割(Polyp Segmentation)作为一种计算机视觉任务，可以自动地将图像或视频中的息肉部位分割出来，能够在结肠镜检查过程中降低息肉的漏检率。
但是，大多数息肉分割模型都需要大量像素级精确标注的数据集作为训练样本，且其分割精度受设备和数据分布的影响较大，对数据集进行像素级精确标注也需要耗费大量的人力物力。
发明内容
提供该发明内容部分以便以简要的形式介绍构思，这些构思将在后面的具体实施方式部分被详细描述。该发明内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征，也不旨在用于限制所要求的保护的技术方案的范围。
第一方面，本公开提供一种息肉分割模型的训练方法，所述方法包括：
基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
第二方面，本公开提供一种息肉分割方法，所述方法包括：
接收到检测的目标图像；
将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于第一方面所述的息肉分割模型的训练方法进行训练所得的。
第三方面，本公开提供一种息肉分割模型的训练装置，所述装置包括：
特征提取模块，用于基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
第一确定模块，用于确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
第二确定模块，用于确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
更新模块，用于根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
获取模块，用于根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
训练模块，用于根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
第四方面，本公开提供一种息肉分割装置，所述装置包括：
接收模块，用于接收到检测的目标图像；
分割模块，用于将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于第一方面所述的息肉分割模型的训练方法进行训练所得的。
第五方面，本公开提供一种计算机可读介质，其上存储有计算机程序，该程序被处理装置执行时实现第一或第二方面所述方法的步骤。
第六方面，本公开提供一种电子设备，包括：
存储装置，其上存储有计算机程序；
处理装置，用于执行所述存储装置中的所述计算机程序，以实现第一或第二方面所述方法的步骤。
通过上述技术方案，可以基于已标注的源域数据和未标注的目标域数据对息肉分割模型进行训练，可以有效降低人工标注的工作量和技术要求。并且，在息肉分割模型的训练过程中，在对图像进行特征提取时，提取包含多个像素点特征的块特征，可以在降低数据处理量的同时能够分析息肉分割特征上的相关性，而分别实现对源域图像和目标域图像的前景特征增强，以提高息肉目标的判别性，进而提高息肉分割的准确性，解决息肉强伪装性导致的难分割问题，从而保证跨域息肉分割的准确度，适用于息肉前景强伪装性的应用场景。
本公开的其他特征和优点将在随后的具体实施方式部分予以详细说明。
附图说明
结合附图并参考以下具体实施方式，本公开各实施例的上述和其他特征、优点及方面将变得更加明显。贯穿附图中，相同或相似的附图标记表示相同或相似的元素。应当理解附图是示意性的，原件和元素不一定按照比例绘制。在附图中：
图1是根据本公开的一种实施方式提供的一种息肉分割模型的训练方法的流程图；
图2是根据本公开的一种实施方式提供的一种息肉分割模型的训练装置的框图；
图3示出了适于用来实现本公开实施例的电子设备的结构示意图。
具体实施方式
下面将参照附图更详细地描述本公开的实施例。虽然附图中显示了本公开的某些实施例，然而应当理解的是，本公开可以通过各种形式来实现，而且不应该被解释为限于这里阐述的实施例，相反提供这些实施例是为了更加透彻和完整地理解本公开。应当理解的是，本公开的附图及实施例仅用于示例性作用，并非用于限制本公开的保护范围。
应当理解，本公开的方法实施方式中记载的各个步骤可以按照不同的顺序执行，和/或并行执行。此外，方法实施方式可以包括附加的步骤和/或省略执行示出的步骤。本公开的范围在此方面不受限制。
本文使用的术语“包括”及其变形是开放性包括，即“包括但不限于”。术语“基于”是“至少部分地基于”。术语“一个实施例”表示“至少一个实施例”；术语“另一实施例”表示“至少一个另外的实施例”；术语“一些实施例”表示“至少一些实施例”。其他术语的相关定义将在下文描述中给出。
需要注意，本公开中提及的“第一”、“第二”等概念仅用于对不同的装置、模块或单元进行区分，并非用于限定这些装置、模块或单元所执行的功能的顺序或者相互依存关系。
需要注意，本公开中提及的“一个”、“多个”的修饰是示意性而非限制性的，本领域技术人员应当理解，除非在上下文另有明确指出，否则应该理解为“一个或多个”。
本公开实施方式中的多个装置之间所交互的消息或者信息的名称仅用于说明性的目的，而并不是用于对这些消息或信息的范围进行限制。
可以理解的是，在使用本公开各实施例公开的技术方案之前，均应当依据相关法律法规通过恰当的方式对本公开所涉及个人信息的类型、使用范围、使用场景等告知用户并获得用户的授权。
例如，在响应于接收到用户的主动请求时，向用户发送提示信息，以明确地提示用户，其请求执行的操作将需要获取和使用到用户的个人信息。从而，使得用户可以根据提示信息来自主地选择是否向执行本公开技术方案的操作的电子设备、应用程序、服务器或存储介质等软件或硬件提供个人信息。
作为一种可选的但非限定性的实现方式，响应于接收到用户的主动请求，向用户发送提示信息的方式例如可以是弹窗的方式，弹窗中可以以文字的方式呈现提示信息。此外，弹窗中还可以承载供用户选择“同意”或者“不同意”向电子设备提供个人信息的选择控件。
可以理解的是，上述通知和获取用户授权过程仅是示意性的，不对本公开的实现方式构成限定，其它满足相关法律法规的方式也可应用于本公开的实现方式中。
同时，可以理解的是，本技术方案所涉及的数据(包括但不限于数据本身、数据的获取或使用)应当遵循相应法律法规及相关规定的要求。
图1所示，为根据本公开的一种实施方式提供的一种息肉分割模型的训练方法，如图1所示，所述方法可以包括：
在步骤11中，基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征。
其中，息肉分割模型可以基于分割网络实现，如基于Unet网络实现该息肉分割模型，在Unet网络中：对于输入的图像，首先利用卷积进行下采样，然后提取出一层又一层的特征，之后利用这一层又一层的特征，再进行上采样，最后得出一个每个像素点对应其种类的图像，从而实现图像分割。
其中，在息肉分割领域下，其对应的是像素级的预测结果，即针对图像中的每一像素点的分类进行预测。示例地，可以将Unet网络中的倒数第二层特征层提取到的特征作为输出特征。如将源域图像输入Unet网络，将Unet网络中的倒数第二层特征层提取到的特征作为第一源域块特征，如将目标域图像输入Unet网络，将Unet网络中的倒数第二层特征层提取到的特征作为第一目标域块特征。其中，该Unet网络的倒数第二层特征层为对图像进行下采样特征提取后再进行上采样特征提取，但未上采样至与原始图像尺寸相同的特征。因此，通过Unet网络获得的特征图中的每一特征值对应于一个块特征，在该特征图中每一特征值可以包含原始的输入图像中的多个像素点对应的特征。
示例地，可以预先设置块特征的尺寸，即每一所述块特征的特征值对应于输入图像中的n*n的像素点的特征，n可以根据实际应用场景进行设置。相应地，该源域块特征中包括源域图像中的前景特征和背景特征，目标域块特征中包括目标域图像中的前景特征和背景特征。其中，块特征数量可以基于实际应用场景进行设置，本公开对此不作限定。为了便于获得相邻像素点之间的关联性，本实施例中，可以对图像进行图像特征提取，从而获得块特征，在降低数据处理量的同时，还能够保证相邻像素点之间的关联，贴合息肉分割场景。
其中，源域图像可以由具有一定经验的医师对该图像进行标注，以获得相应的息肉分割标签，在息肉分割领域下，该息肉分割标签可以包括各个像素点的分类标签，以用于表示源域图像中哪些像素点对应息肉。
如本公开背景技术中所述，来自于不同的模态、设备或医院上的数据其对应的分布可能不同，则若以源域数据集中的图像训练的息肉分割模型可能难以直接对目标域下的图像进行检测。然后在息肉分割领域下，由于数据量相对较小，标注要求较高等问题，对训练图像进行标注的工作量较大。因此，在本公开中可以同时基于源域数据集和目标域数据集中的图像对模型进行训练，无需对目标域数据集中的数据进行标注的情况下以使得训练完成的息肉分割模型能够适应于目标域下的图像中息肉分割。
如源域数据集可以是来自于Hs1医院的图像数据，其标注的相应的息肉标签，目标域数据集可以是来自于Hs2医院的图像数据。
在步骤12中，确定第一源域块特征中的源域前景块特征和第一目标域块特征中的目标域前景块特征。
其中，在源域图像和目标域图像中，息肉所处的位置通常位于图像中的前景位置，相应地，在该步骤中，源域前景块特征用于表示源域图像中预测出的对应于息肉的块特征，目标域前景块特征则用于表示目标域图像中预测出的对应于息肉的块特征，以便对息肉进行准确分割。
在步骤13中，确定源域前景块特征和目标域前景块特征对应的连接关系。
其中，针对同一图像而言，其中各个像素点的特征是有所关联的，而针对源域和目标域中的不同图像而言，可以通过确定其分别对应的像素的特征之间关联性，以实现对目标域下的跨域图像的准确分割。则在该实施例中，通过确定该连接关系以对源域图像和目标域图像对应的特征之间的关联性进行分析，通过构建连接关系进行不同域下的图像特征之间的分布对齐。
在步骤14中，根据连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征。
其中，在息肉分割场景下，由于息肉的颜色和质地与周围正常组织非常相似，二者之间的对比度低，使得息肉有较大的伪装特性，难以对息肉边缘进行准确检测。在步骤中，连接关系用于表示源域的前景特征和目标域的前景特征对应的关联性，从而可以基于该连接关系对第一源域块特征和第一目标域块特征进行前景特征增强，以便于突出息肉对应的前景特征与其他区域的对比度，从而可以在一定程度提高源域图像对应的特征图像中的前景特征表达，以适用于息肉前景强伪装性的应用场景。
在步骤15中，根据第二源域块特征获得源域图像对应的预测分割结果，该步骤中可以基于Unet网络中的分类器进行预测，获得该预测分割结果，其中，该预测分割结果包含源域图像中各个像素点对应的分类，以指示其对应于息肉分类还是非息肉分类，便于息肉边缘分割。
在步骤16中，根据源域图像对应的预测分割结果和息肉分割标签、以及第二源域块特征和第二目标域块特征，确定息肉分割模型的目标损失，并根据目标损失对息肉分割模型进行训练。
其中，基于根据源域图像对应的预测分割结果和息肉分割标签可以确定该息肉分割模型的分割准确度，基于第二源域块特征和第二目标域块特征，可以通过约束两者分布接近以使得源域特征和目标域特征对齐，而提高息肉在跨域图像下的分割准确度，则目标损失中可以包含息肉分割模型的准确度，以及源域块特征和目标域块特征分布的距离程度。
示例地，可以在目标损失小于损失阈值时停止训练，也可以在训练次数达到次数阈值时停止训练，反之则基于该目标损失通过梯度下降的方式对息肉分割模型中的参数进行更新，以再次通过上文所述的步骤进行重复训练，直至模型训练完成。
由此，通过上述技术方案，可以基于已标注的源域数据和未标注的目标域数据对息肉分割模型进行训练，可以有效降低人工标注的工作量和技术要求。并且，在息肉分割模型的训练过程中，在对图像进行特征提取时，提取包含多个像素点特征的块特征，可以在降低数据处理量的同时能够分析息肉分割特征上的相关性，而分别实现对源域图像和目标域图像的前景特征增强，以提高息肉目标的判别性，进而提高息肉分割的准确性，解决息肉强伪装性导致的难分割问题，从而保证跨域息肉分割的准确度，适用于息肉前景强伪装性的应用场景。
在一种可能的实施例中，在步骤12中确定第一源域块特征中的源域前景块特征和第一目标域块特征中的目标域前景块特征的示例性实现方式如下，该步骤可以包括：
对所述息肉分割标签对应的位置进行特征切分，获得分割标签块特征。
其中，源域图像对应的息肉分割标签为像素级的标签，即其是针对每一像素点的标签，则该步骤中可以对源域图像基于该息肉分割标签进行切分，以将对应于前景特征的像素点进行特征切分，获得多个分割标签块特征，其中所述分割标签块特征对应的像素点尺寸与每一所述第一源域块特征对应的像素点尺寸相同。
根据所述分割标签块特征确定所述第一源域块特征对应的候选前景块特征集合。
示例地，第一源域块特征为fs，则可以将第一源域块特征中与分割标签块特征对应的块特征作为候选前景块特征集合fs fg，即fs fg∈fs。
针对所述候选前景块特征集合中的每一所述候选前景块特征，确定所述第一目标域块特征中与所述候选前景块特征对应的目标域匹配前景块特征，并确定所述目标域匹配前景块特征在所述第一源域块特征中的源域匹配前景块特征。
其中，针对第i个候选前景块特征fs fg,i，可以分别计算该候选前景块特征fs fg,i与第一目标域块特征ft中的每一个块特征ft j之间的余弦相似度，并将对应余弦相似度最大的块特征确定为所述候选前景块特征fs fg,i对应的目标域匹配前景块特征，目标域匹配前景块特征的下标表示如下：
j'＝argmaxcos(fs fg,i,ft j)
j∈{1,2,...,Ht×Wt}
其中，Ht和Wt用于表示第一目标域块特征ft的高和宽，即第一目标域块特征中的每一个特征值对应于一个块特征，与候选前景块特征fs fg,i对应的目标域匹配前景块特征为第一目标域块特征中的第j'个块特征。
进一步地，针对该目标域匹配前景块特征，即第一目标域块特征中的第j'个块特征，可以进一步计算其与第一源域块特征中的每一块特征之间的余弦相似度，并将对应余弦相似度最大的块特征确定为所述目标域匹配前景块特征在所述第一源域块特征中的源域匹配前景块特征，示例地，可以进一步通过类似的方式确定出第一目标域块特征中的第j'个块特征ft j'在第一源域块特征中的源域匹配前景块特征，如该源域匹配前景块特征为第一源域块特征中的第i'个块特征。
之后，若所述源域匹配前景块特征属于所述候选前景块特征集合，将所述候选前景块特征确定为所述源域前景块特征，并将所述目标域匹配前景块特征确定为所述目标域前景块特征。
示例地，若第一源域块特征中的第i'个块特征属于fs fg，此时可以认为该第一源域块特征中的候选前景块特征fs fg,i为源域前景块特征，同时可以确定第一目标域块特征中的特征ft j'为目标域前景块特征。
由此，通过上述技术方案，可以通过源域块特征和目标域块特征对应的特征相似度，对源域前景块特征和目标域前景块特征进行筛选，从而从源域图像和目标域图像中确定出具有代表性的块特征，以在未对目标域图像进行标注的情况下，准确确定出目标域前景块特征，提高该方法的应用范围。
在一种可能的实施例中，步骤13中确定源域前景块特征和目标域前景块特征对应的连接关系的示例性实现方式可以包括：
根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，其中，所述目标类型包括源域类型和目标域类型，所述域连接关系包括所述源域类型下的源域前景块特征之间的源域连接关系和所述目标域类型下的所述目标域前景块特征之间的目标域连接关系。
示例地，所述根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，包括：
针对每一所述目标类型下各个域前景块特征，确定所述目标类型下的所述域前景块特征与各个候选域前景块特征之间的相似度，所述候选域前景块特征与所述域前景块特征不同，且对应于同一目标类型。示例地，源域类型下的域前景块特征即所述源域前景块特征，目标域类型下的域前景块特征即所述目标域前景块特征。
其中，以第一源域块特征中的每一源域前景块特征为例，可以分别计算该源域前景块特征与其他的每一源域前景块特征之间的相似度，如可以基于特征进行余弦相似度计算，余弦相似度的计算方式为现有技术，在此不再赘述。
若所述相似度大于或等于预设阈值，则确定所述域前景块特征与所述候选域前景块特征之间具备连接关系，所述目标类型下的域连接关系包括所述目标类型下各个域前景块特征对应的接连关系。
其中，预设阈值可以基于实际应用场景进行设置，本公开对此不进行限定。以源域前景块特征为例，若源域前景块特征与候选源域前景块特征之间的相似度大于或等于预设阈值，则该源域前景块特征和候选源域前景块特征的特征相近，此时可以构建源域前景块特征与所述候选源域前景块特征之间的连接关系，以挖掘源域图像中的特征之间的关联性，以及目标域图像中的特征之间的关联性。
示例地，源域连接关系可以通过m*m的邻接矩阵表示，矩阵中元素取值为1表示该位置对应的两个源域前景块特征之间具备连接关系，矩阵中元素取值为0表示该位置对应的两个源域前景块特征之间不具备连接关系。
相应地，可以通过上述类似的方式对目标域类型下的各个目标域前景块特征计算与其他目标域前景块特征之间的相似度，进而确定目标域连接关系，在此不再赘述。
相应地，步骤13中确定源域前景块特征和目标域前景块特征对应的连接关系的步骤还可以包括：
基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系。
示例地，所述基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系可以包括：
通过以下公式确定第i个源域前景块特征和第j个目标域前景块特征之间的域间连接关系值
Figure BDA0003812292600000111
Figure BDA0003812292600000121
其中，
Figure BDA0003812292600000122
用于表示源域图像s中的第i个源域前景块特征；
Figure BDA0003812292600000123
用于表示目标域图像t中的第j个目标域前景块特征；
σ用于表示调节超参数；
相应地，所述域间连接关系包括每一所述源域前景块特征和每一所述目标域前景块特征之间的域间连接关系值。
其中，源域前景块特征用于表示源域图像中的前景特征，目标域前景块特征用于表示目标域图像中的前景特征，通过上述公式可以确定源域和目标域中的前景特征之间的相关性，既可以便于后续基于该域间连接关系进行前景特征增强，又可以为缩短源域和目标域中的前景特征的特征分布差异提供可靠的数据支持，保证跨域场景下对目标域中的息肉特征的准确识别，提高目标域下息肉分割的准确性。
在一种可能的实施例中，所述根据所述连接关系对第一块特征进行更新，获得第二块特征的示例性实现方式如下，该步骤可以包括：
根据所述目标类型下的域连接关系和至少一个注意力特征层对所述目标类型下的第一块特征进行卷积处理，获得所述目标类型下的卷积块特征，其中，所述目标类型下的注意力特征层中对应的注意力权重为基于所述目标类型下的域连接关系确定的。
其中，所述目标类型下的注意力特征层中对应的注意力权重基于所述目标类型下的域连接关系通过如下公式确定：
Figure BDA0003812292600000124
其中，
Figure BDA0003812292600000125
用于表示第l个注意力特征层所述目标类型下的第i个域前景块特征和第j个域前景块特征之间的注意力权重，其中，所述目标类型下的第i个域前景块特征和第j个域前景块特征之间具有连接关系；
Aij用于表示所述目标类型下的第i个域前景块特征和第j个域前景块特征之间的连接关系值；
Figure BDA0003812292600000131
用于表示所述目标类型下的第i个域前景块特征在第l个注意力特征层中的特征；
Ni用于表示所述目标类型下的第i个域前景块特征具有连接关系的域前景块特征的集合；
W用于表示可学习的特征变换操作；
Figure BDA0003812292600000132
用于表示可学习的权重向量；
||用于表示特征拼接操作。
相应地，在确定出当前的注意力特征层对应的注意力权重后，则可以通过以下公式获得当前的注意力特征层输出的注意力特征：
Figure BDA0003812292600000133
其中，
Figure BDA0003812292600000134
用于表示所述目标类型下的第i个域前景块特征在第l层的特征
Figure BDA0003812292600000135
进行注意力特征提取后、在第l+1层的注意力特征，其中第l层的特征为所述目标类型下的第一块特征；
δ()用于表示非线性映射。
其中，每一注意力特征层均可以通过上述方式确定当前层对应的注意力权重，从而基于注意力权重对输入特征进行增强更新，最后一层注意力特征层对应的注意力特征可以作为该所述目标类型下的卷积块特征。
由此，通过上述技术方案，可以基于目标类型下的域连接关系确定注意力特征提取时的注意力权重，进而通过注意力权重进行特征更新，完成基于多层图注意力机制的前景特征增强。
之后，则可以根据所述域间连接关系分别对每一所述目标类型下的所述卷积块特征进行多层卷积特征提取，获得所述目标类型下的第二块特征，其中，所述第一源域块特征和所述第一目标域块特征对应的卷积特征提取的层数相同。
其中，卷积特征提取的层数可以根据实际应用场景进行设置，本公开对此不做限定。示例地，针对每一目标类型，可以通过以下公式基于所述域间连接关系，对所述卷积块特征进行多层卷积特征提取，获得所述第二块特征：
Figure BDA0003812292600000141
其中，Ast用于源域图像s的源域前景块特征和目标域图像t中的目标域前景块特征之间的域间连接关系对应的矩阵；
I用于表示单位矩阵；
D'用于表示Ast的度矩阵，其中，D'ii＝∑jA'ij；
H(l)用于表示第l层对应的卷积特征，其中第0层对应的卷积特征为所述卷积块特征；
Hl+1用于表示第l+1层对应的卷积特征；
W(l)用于表示第l层对应的权重参数；
η()用于表示非线性激活函数，如RELU函数。
在该技术方案中，在基于目标类型下的域连接关系分别对提取出的块特征进行更新后，可以进一步基于域间连接关系对更新后的特征进行再次更新。该过程中，通过确定Ast的度矩阵可以确定第i个源域前景块特征与目标域前景块特征之间综合相关度，从而实现源域特征与目标域特征的整体关联，确定出源域特征和目标域特征的整体相关性，便于约束获得的第二源域块特征和第二目标域块特征的一致性。
在一种可能的实施例中，所述根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，包括：
根据所述预测分割结果中的预测分割信息和所述息肉分割标签，确定分割损失。其中，可以根据Unet网络中的损失函数进行计算，如交叉熵损失函数。
根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失。
示例地，通过以下公式根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失：
Figure BDA0003812292600000151
其中，Lcst用于表示所述特征对齐损失；
Ns用于表示所述源域前景块特征的数量，即从第一源域块特征的各个特征值中确定出的对应于源域前景块特征的特征值的数量；
Nt用于表示所述目标域前景块特征的数量，即从第一目标域块特征的各个特征值中确定出的对应于目标域前景块特征的特征值的数量；
xi s用于表示所述第二源域块特征中第i个源域前景块特征；
xj t用于表示所述第二目标域块特征中第j个目标域前景块特征。
之后，可以根据所述分割损失和所述特征对齐损失，确定所述目标损失。
示例地，可以基于分割损失和特征对齐损失对应的权重对其进行加权求和，以获得该目标损失。由此，通过上述技术方案，可以在目标损失中既可以包含用于约束息肉分割结果准确度的损失，也可以包含用于约束源域的图像像素特征和目标域的图像像素特征之间的特征分布的损失，使得在息肉分割模型的训练过程中提高分割结果准确度的同时，能够使得该模型适用于目标域下的息肉图像的分割场景，提高息肉分割模型对跨域状态下息肉分割预测的准确性，拓宽息肉分割模型的使用范围。
本公开还提供一种息肉分割方法，所述方法包括：
接收到检测的目标图像，其中该目标图像可以是从实时检测的内窥镜中进行采集所得的图像。
将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割检测结果，其中，所述息肉分割模型为基于上文任一所述的息肉分割模型的训练方法进行训练所得的。
通过上述技术方案，可以基于训练完成的息肉分割模型对目标图像进行息肉分割，保证息肉分割的准确度和效率。并且该息肉分割模型为基于源域图像和目标域图像作为训练样本进行训练的，在该过程中可以增强息肉前景的特征表达，使得训练完成的息肉分割模型能够降低息肉前景特征的强伪装性所导致的检测误差，进一步拓宽息肉分割方法的使用范围和息肉分割结果的准确性。
本公开还提供一种息肉分割模型的训练装置，如图2所示，所述装置10包括：
特征提取模块101，用于基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
第一确定模块102，用于确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
第二确定模块103，用于确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
更新模块104，用于根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
获取模块105，用于根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
训练模块106，用于根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
可选地，所述第一确定模块包括：
切分子模块，用于对所述息肉分割标签对应的位置进行特征切分，获得分割标签块特征；
第一确定子模块，用于根据所述分割标签块特征确定所述第一源域块特征对应的候选前景块特征集合；
第二确定子模块，用于针对所述候选前景块特征集合中的每一所述候选前景块特征，确定所述第一目标域块特征中与所述候选前景块特征对应的目标域匹配前景块特征，并确定所述目标域匹配前景块特征在所述第一源域块特征中的源域匹配前景块特征；
第三确定子模块，用于若所述源域匹配前景块特征属于所述候选前景块特征集合，将所述候选前景块特征确定为所述源域前景块特征，并将所述目标域匹配前景块特征确定为所述目标域前景块特征。
可选地，所述第二确定模块包括：
第四确定子模块，用于根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，其中，所述目标类型包括源域类型和目标域类型，所述域连接关系包括所述源域类型下的源域前景块特征之间的源域连接关系和所述目标域类型下的所述目标域前景块特征之间的目标域连接关系；
第五确定子模块，用于基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系。
可选地，所述第四确定子模块，包括：
第六确定子模块，用于针对每一所述目标类型下各个域前景块特征，确定所述目标类型下的所述域前景块特征与各个候选域前景块特征之间的相似度，所述候选域前景块特征与所述域前景块特征不同，且对应于同一目标类型；
第七确定子模块，用于若所述相似度大于或等于预设阈值，则确定所述域前景块特征与所述候选域前景块特征之间具备连接关系，所述目标类型下的域连接关系包括所述目标类型下各个域前景块特征对应的接连关系。
可选地，所述第五确定子模块用于：
通过以下公式确定第i个源域前景块特征和第j个目标域前景块特征之间的域间连接关系值
Figure BDA0003812292600000171
Figure BDA0003812292600000181
其中，
Figure BDA0003812292600000182
用于表示源域图像s中的第i个源域前景块特征；
Figure BDA0003812292600000183
用于表示目标域图像t中的第j个目标域前景块特征；
σ用于表示调节超参数；
所述域间连接关系包括每一所述源域前景块特征和每一所述目标域前景块特征之间的域间连接关系值。
可选地，所述更新模块包括：
第一更新子模块，用于根据所述目标类型下的域连接关系和至少一个注意力特征层对所述目标类型下的第一块特征进行卷积处理，获得所述目标类型下的卷积块特征，其中，所述目标类型下的注意力特征层中对应的注意力权重为基于所述目标类型下的域连接关系确定的；
第二更新子模块，用于根据所述域间连接关系分别对每一所述目标类型下的所述卷积块特征进行多层卷积特征提取，获得所述目标类型下的第二块特征，其中，所述第一源域块特征和所述第一目标域块特征对应的卷积特征提取的层数相同。
可选地，所述训练模块包括：
第八确定子模块，用于根据所述预测分割结果中的预测分割信息和所述息肉分割标签，确定分割损失；
第九确定子模块，用于根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失；
第十确定子模块，用于根据所述分割损失和所述特征对齐损失，确定所述目标损失。
可选地，所述第九确定子模块通过以下公式根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失：
Figure BDA0003812292600000184
其中，Lcst用于表示所述特征对齐损失；
Ns用于表示所述源域前景块特征的数量；
Nt用于表示所述目标域前景块特征的数量；
xi s用于表示所述第二源域块特征中第i个源域前景块特征；
xj t用于表示所述第二目标域块特征中第j个目标域前景块特征。
本公开还提供一种息肉分割装置，所述装置包括：
接收模块，用于接收到检测的目标图像；
分割模块，用于将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于上文任一所述的息肉分割模型的训练方法进行训练所得的。
下面参考图3，其示出了适于用来实现本公开实施例的电子设备600的结构示意图。本公开实施例中的终端设备可以包括但不限于诸如移动电话、笔记本电脑、数字广播接收器、PDA(个人数字助理)、PAD(平板电脑)、PMP(便携式多媒体播放器)、车载终端(例如车载导航终端)等等的移动终端以及诸如数字TV、台式计算机等等的固定终端。图3示出的电子设备仅仅是一个示例，不应对本公开实施例的功能和使用范围带来任何限制。
如图3所示，电子设备600可以包括处理装置(例如中央处理器、图形处理器等)601，其可以根据存储在只读存储器(ROM)602中的程序或者从存储装置608加载到随机访问存储器(RAM)603中的程序而执行各种适当的动作和处理。在RAM 603中，还存储有电子设备600操作所需的各种程序和数据。处理装置601、ROM 602以及RAM 603通过总线604彼此相连。输入/输出(I/O)接口605也连接至总线604。
通常，以下装置可以连接至I/O接口605：包括例如触摸屏、触摸板、键盘、鼠标、摄像头、麦克风、加速度计、陀螺仪等的输入装置606；包括例如液晶显示器(LCD)、扬声器、振动器等的输出装置607；包括例如磁带、硬盘等的存储装置608；以及通信装置609。通信装置609可以允许电子设备600与其他设备进行无线或有线通信以交换数据。虽然图3示出了具有各种装置的电子设备600，但是应理解的是，并不要求实施或具备所有示出的装置。可以替代地实施或具备更多或更少的装置。
特别地，根据本公开的实施例，上文参考流程图描述的过程可以被实现为计算机软件程序。例如，本公开的实施例包括一种计算机程序产品，其包括承载在非暂态计算机可读介质上的计算机程序，该计算机程序包含用于执行流程图所示的方法的程序代码。在这样的实施例中，该计算机程序可以通过通信装置609从网络上被下载和安装，或者从存储装置608被安装，或者从ROM 602被安装。在该计算机程序被处理装置601执行时，执行本公开实施例的方法中限定的上述功能。
需要说明的是，本公开上述的计算机可读介质可以是计算机可读信号介质或者计算机可读存储介质或者是上述两者的任意组合。计算机可读存储介质例如可以是——但不限于——电、磁、光、电磁、红外线、或半导体的系统、装置或器件，或者任意以上的组合。计算机可读存储介质的更具体的例子可以包括但不限于：具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机访问存储器(RAM)、只读存储器(ROM)、可擦式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑磁盘只读存储器(CD-ROM)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本公开中，计算机可读存储介质可以是任何包含或存储程序的有形介质，该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。而在本公开中，计算机可读信号介质可以包括在基带中或者作为载波一部分传播的数据信号，其中承载了计算机可读的程序代码。这种传播的数据信号可以采用多种形式，包括但不限于电磁信号、光信号或上述的任意合适的组合。计算机可读信号介质还可以是计算机可读存储介质以外的任何计算机可读介质，该计算机可读信号介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。计算机可读介质上包含的程序代码可以用任何适当的介质传输，包括但不限于：电线、光缆、RF(射频)等等，或者上述的任意合适的组合。
在一些实施方式中，客户端、服务器可以利用诸如HTTP(HyperText TransferProtocol，超文本传输协议)之类的任何当前已知或未来研发的网络协议进行通信，并且可以与任意形式或介质的数字数据通信(例如，通信网络)互连。通信网络的示例包括局域网(“LAN”)，广域网(“WAN”)，网际网(例如，互联网)以及端对端网络(例如，ad hoc端对端网络)，以及任何当前已知或未来研发的网络。
上述计算机可读介质可以是上述电子设备中所包含的；也可以是单独存在，而未装配入该电子设备中。
上述计算机可读介质承载有一个或者多个程序，当上述一个或者多个程序被该电子设备执行时，使得该电子设备：基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；根据所述第二源域块特征获得所述源域图像对应的预测分割结果；根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
或者，上述计算机可读介质承载有一个或者多个程序，当上述一个或者多个程序被该电子设备执行时，使得该电子设备：接收到检测的目标图像；将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于任一所述的息肉分割模型的训练方法进行训练所得的。
可以以一种或多种程序设计语言或其组合来编写用于执行本公开的操作的计算机程序代码，上述程序设计语言包括但不限于面向对象的程序设计语言—诸如Java、Smalltalk、C++，还包括常规的过程式程序设计语言——诸如“C”语言或类似的程序设计语言。程序代码可以完全地在用户计算机上执行、部分地在用户计算机上执行、作为一个独立的软件包执行、部分在用户计算机上部分在远程计算机上执行、或者完全在远程计算机或服务器上执行。在涉及远程计算机的情形中，远程计算机可以通过任意种类的网络——包括局域网(LAN)或广域网(WAN)——连接到用户计算机，或者，可以连接到外部计算机(例如利用因特网服务提供商来通过因特网连接)。
附图中的流程图和框图，图示了按照本公开各种实施例的系统、方法和计算机程序产品的可能实现的体系架构、功能和操作。在这点上，流程图或框图中的每个方框可以代表一个模块、程序段、或代码的一部分，该模块、程序段、或代码的一部分包含一个或多个用于实现规定的逻辑功能的可执行指令。也应当注意，在有些作为替换的实现中，方框中所标注的功能也可以以不同于附图中所标注的顺序发生。例如，两个接连地表示的方框实际上可以基本并行地执行，它们有时也可以按相反的顺序执行，这依所涉及的功能而定。也要注意的是，框图和/或流程图中的每个方框、以及框图和/或流程图中的方框的组合，可以用执行规定的功能或操作的专用的基于硬件的系统来实现，或者可以用专用硬件与计算机指令的组合来实现。
描述于本公开实施例中所涉及到的模块可以通过软件的方式实现，也可以通过硬件的方式来实现。其中，模块的名称在某种情况下并不构成对该模块本身的限定，例如，特征提取模块还可以被描述为“基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征的模块”。
本文中以上描述的功能可以至少部分地由一个或多个硬件逻辑部件来执行。例如，非限制性地，可以使用的示范类型的硬件逻辑部件包括：现场可编程门阵列(FPGA)、专用集成电路(ASIC)、专用标准产品(ASSP)、片上系统(SOC)、复杂可编程逻辑设备(CPLD)等等。
在本公开的上下文中，机器可读介质可以是有形的介质，其可以包含或存储以供指令执行系统、装置或设备使用或与指令执行系统、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体系统、装置或设备，或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦除可编程只读存储器(EPROM或快闪存储器)、光纤、便捷式紧凑盘只读存储器(CD-ROM)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
根据本公开的一个或多个实施例，示例1提供了一种息肉分割模型的训练方法，其中，所述方法包括：
基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
根据本公开的一个或多个实施例，示例2提供了示例1的方法，其中，所述确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征，包括：
对所述息肉分割标签对应的位置进行特征切分，获得分割标签块特征；
根据所述分割标签块特征确定所述第一源域块特征对应的候选前景块特征集合；
针对所述候选前景块特征集合中的每一所述候选前景块特征，确定所述第一目标域块特征中与所述候选前景块特征对应的目标域匹配前景块特征，并确定所述目标域匹配前景块特征在所述第一源域块特征中的源域匹配前景块特征；
若所述源域匹配前景块特征属于所述候选前景块特征集合，将所述候选前景块特征确定为所述源域前景块特征，并将所述目标域匹配前景块特征确定为所述目标域前景块特征。
根据本公开的一个或多个实施例，示例3提供了示例1的方法，其中，所述确定所述源域前景块特征和所述目标域前景块特征对应的连接关系，包括：
根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，其中，所述目标类型包括源域类型和目标域类型，所述域连接关系包括所述源域类型下的源域前景块特征之间的源域连接关系和所述目标域类型下的所述目标域前景块特征之间的目标域连接关系；
基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系。
根据本公开的一个或多个实施例，示例4提供了示例3的方法，其中，所述根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，包括：
针对每一所述目标类型下各个域前景块特征，确定所述目标类型下的所述域前景块特征与各个候选域前景块特征之间的相似度，所述候选域前景块特征与所述域前景块特征不同，且对应于同一目标类型；
若所述相似度大于或等于预设阈值，则确定所述域前景块特征与所述候选域前景块特征之间具备连接关系，所述目标类型下的域连接关系包括所述目标类型下各个域前景块特征对应的接连关系。
根据本公开的一个或多个实施例，示例5提供了示例3的方法，其中，所述基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系，包括：
通过以下公式确定第i个源域前景块特征和第j个目标域前景块特征之间的域间连接关系值
Figure BDA0003812292600000251
Figure BDA0003812292600000252
其中，
Figure BDA0003812292600000253
用于表示源域图像s中的第i个源域前景块特征；
Figure BDA0003812292600000254
用于表示目标域图像t中的第j个目标域前景块特征；
σ用于表示调节超参数；
所述域间连接关系包括每一所述源域前景块特征和每一所述目标域前景块特征之间的域间连接关系值。
根据本公开的一个或多个实施例，示例6提供了示例3的方法，其中，所述根据所述连接关系对第一块特征进行更新，获得第二块特征，包括：
根据所述目标类型下的域连接关系和至少一个注意力特征层对所述目标类型下的第一块特征进行卷积处理，获得所述目标类型下的卷积块特征，其中，所述目标类型下的注意力特征层中对应的注意力权重为基于所述目标类型下的域连接关系确定的；
根据所述域间连接关系分别对每一所述目标类型下的所述卷积块特征进行多层卷积特征提取，获得所述目标类型下的第二块特征，其中，所述第一源域块特征和所述第一目标域块特征对应的卷积特征提取的层数相同。
根据本公开的一个或多个实施例，示例7提供了示例1的方法，其中，所述根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，包括：
根据所述预测分割结果中的预测分割信息和所述息肉分割标签，确定分割损失；
根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失；
根据所述分割损失和所述特征对齐损失，确定所述目标损失。
根据本公开的一个或多个实施例，示例8提供了示例7的方法，其中，通过以下公式根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失：
Figure BDA0003812292600000261
其中，Lcst用于表示所述特征对齐损失；
Ns用于表示所述源域前景块特征的数量；
Nt用于表示所述目标域前景块特征的数量；
xi s用于表示所述第二源域块特征中第i个源域前景块特征；
xj t用于表示所述第二目标域块特征中第j个目标域前景块特征。
根据本公开的一个或多个实施例，示例9提供了一种息肉分割方法，其中，所述方法包括：
接收到检测的目标图像；
将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于示例1-8中任一项所述的息肉分割模型的训练方法进行训练所得的。
根据本公开的一个或多个实施例，示例10提供了一种息肉分割模型的训练装置，其中，所述装置包括：
特征提取模块，用于基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
第一确定模块，用于确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
第二确定模块，用于确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
更新模块，用于根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
获取模块，用于根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
训练模块，用于根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
根据本公开的一个或多个实施例，示例11提供了一种息肉分割装置，其中，所述装置包括：
接收模块，用于接收到检测的目标图像；
分割模块，用于将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于示例1-8中任一所述的息肉分割模型的训练方法进行训练所得的。
根据本公开的一个或多个实施例，示例12提供了一种计算机可读介质，其上存储有计算机程序，该程序被处理装置执行时实现示例1-9中任一项所述方法的步骤。
根据本公开的一个或多个实施例，示例13提供了一种电子设备，包括：
存储装置，其上存储有计算机程序；
处理装置，用于执行所述存储装置中的所述计算机程序，以实现示例1-9中任一项所述方法的步骤。
以上描述仅为本公开的较佳实施例以及对所运用技术原理的说明。本领域技术人员应当理解，本公开中所涉及的公开范围，并不限于上述技术特征的特定组合而成的技术方案，同时也应涵盖在不脱离上述公开构思的情况下，由上述技术特征或其等同特征进行任意组合而形成的其它技术方案。例如上述特征与本公开中公开的(但不限于)具有类似功能的技术特征进行互相替换而形成的技术方案。
此外，虽然采用特定次序描绘了各操作，但是这不应当理解为要求这些操作以所示出的特定次序或以顺序次序执行来执行。在一定环境下，多任务和并行处理可能是有利的。同样地，虽然在上面论述中包含了若干具体实现细节，但是这些不应当被解释为对本公开的范围的限制。在单独的实施例的上下文中描述的某些特征还可以组合地实现在单个实施例中。相反地，在单个实施例的上下文中描述的各种特征也可以单独地或以任何合适的子组合的方式实现在多个实施例中。
尽管已经采用特定于结构特征和/或方法逻辑动作的语言描述了本主题，但是应当理解所附权利要求书中所限定的主题未必局限于上面描述的特定特征或动作。相反，上面所描述的特定特征和动作仅仅是实现权利要求书的示例形式。关于上述实施例中的装置，其中各个模块执行操作的具体方式已经在有关该方法的实施例中进行了详细描述，此处将不做详细阐述说明。

Claims (13)

1.一种息肉分割模型的训练方法，其特征在于，所述方法包括：
基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
2.根据权利要求1所述的方法，其特征在于，所述确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征，包括：
对所述息肉分割标签对应的位置进行特征切分，获得分割标签块特征；
根据所述分割标签块特征确定所述第一源域块特征对应的候选前景块特征集合；
针对所述候选前景块特征集合中的每一所述候选前景块特征，确定所述第一目标域块特征中与所述候选前景块特征对应的目标域匹配前景块特征，并确定所述目标域匹配前景块特征在所述第一源域块特征中的源域匹配前景块特征；
若所述源域匹配前景块特征属于所述候选前景块特征集合，将所述候选前景块特征确定为所述源域前景块特征，并将所述目标域匹配前景块特征确定为所述目标域前景块特征。
3.根据权利要求1所述的方法，其特征在于，所述确定所述源域前景块特征和所述目标域前景块特征对应的连接关系，包括：
根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，其中，所述目标类型包括源域类型和目标域类型，所述域连接关系包括所述源域类型下的源域前景块特征之间的源域连接关系和所述目标域类型下的所述目标域前景块特征之间的目标域连接关系；
基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系。
4.根据权利要求3所述的方法，其特征在于，所述根据每一目标类型下的域前景块特征，确定所述目标类型下的域连接关系，包括：
针对每一所述目标类型下各个域前景块特征，确定所述目标类型下的所述域前景块特征与各个候选域前景块特征之间的相似度，所述候选域前景块特征与所述域前景块特征不同，且对应于同一目标类型；
若所述相似度大于或等于预设阈值，则确定所述域前景块特征与所述候选域前景块特征之间具备连接关系，所述目标类型下的域连接关系包括所述目标类型下各个域前景块特征对应的接连关系。
5.根据权利要求3所述的方法，其特征在于，所述基于各个所述源域前景块特征和所述目标域前景块特征，确定所述源域前景块特征和所述目标域前景块特征之间的域间连接关系，包括：
通过以下公式确定第i个源域前景块特征和第j个目标域前景块特征之间的域间连接关系值
Figure FDA0003812292590000031
Figure FDA0003812292590000032
其中，
Figure FDA0003812292590000033
用于表示源域图像s中的第i个源域前景块特征；
Figure FDA0003812292590000034
用于表示目标域图像t中的第j个目标域前景块特征；
σ用于表示调节超参数；
所述域间连接关系包括每一所述源域前景块特征和每一所述目标域前景块特征之间的域间连接关系值。
6.根据权利要求3所述的方法，其特征在于，所述根据所述连接关系对第一块特征进行更新，获得第二块特征，包括：
根据所述目标类型下的域连接关系和至少一个注意力特征层对所述目标类型下的第一块特征进行卷积处理，获得所述目标类型下的卷积块特征，其中，所述目标类型下的注意力特征层中对应的注意力权重为基于所述目标类型下的域连接关系确定的；
根据所述域间连接关系分别对每一所述目标类型下的所述卷积块特征进行多层卷积特征提取，获得所述目标类型下的第二块特征，其中，所述第一源域块特征和所述第一目标域块特征对应的卷积特征提取的层数相同。
7.根据权利要求1所述的方法，其特征在于，所述根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，包括：
根据所述预测分割结果中的预测分割信息和所述息肉分割标签，确定分割损失；
根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失；
根据所述分割损失和所述特征对齐损失，确定所述目标损失。
8.根据权利要求7所述的方法，其特征在于，通过以下公式根据所述第二源域块特征和所述第二目标域块特征，确定特征对齐损失：
Figure FDA0003812292590000041
其中，Lcst用于表示所述特征对齐损失；
Ns用于表示所述源域前景块特征的数量；
Nt用于表示所述目标域前景块特征的数量；
xi s用于表示所述第二源域块特征中第i个源域前景块特征；
xj t用于表示所述第二目标域块特征中第j个目标域前景块特征。
9.一种息肉分割方法，其特征在于，所述方法包括：
接收到检测的目标图像；
将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于权利要求1-8中任一项所述的息肉分割模型的训练方法进行训练所得的。
10.一种息肉分割模型的训练装置，其特征在于，所述装置包括：
特征提取模块，用于基于息肉分割模型对源域数据集中的源域图像和目标域数据集中的目标域图像进行图像特征提取，获得第一源域块特征和第一目标域块特征，其中，所述源域数据集中的图像标注有息肉分割标签，所述目标域数据集中的图像未标注标签，所述源域数据集与所述目标域数据集的数据分布不同，所述第一源域块特征中的每一特征值对应于所述源域图像中n*n的像素点的特征，所述第一目标域块特征中的每一特征值对应于所述目标域图像中n*n的像素点的特征；
第一确定模块，用于确定所述第一源域块特征中的源域前景块特征和所述第一目标域块特征中的目标域前景块特征；
第二确定模块，用于确定所述源域前景块特征和所述目标域前景块特征对应的连接关系；
更新模块，用于根据所述连接关系对第一块特征进行更新，获得第二块特征，其中，所述第一块特征包括源域类型下的第一源域块特征和目标域类型下的所述第一目标域块特征，所述第二块特征包括根据所述连接关系对所述第一源域块特征更新后得到的第二源域块特征和根据所述连接关系对所述第一目标域块特征更新后得到的第二目标域块特征；
获取模块，用于根据所述第二源域块特征获得所述源域图像对应的预测分割结果；
训练模块，用于根据所述源域图像对应的预测分割结果和所述息肉分割标签、以及所述第二源域块特征和所述第二目标域块特征，确定所述息肉分割模型的目标损失，并根据所述目标损失对所述息肉分割模型进行训练。
11.一种息肉分割装置，其特征在于，所述装置包括：
接收模块，用于接收到检测的目标图像；
分割模块，用于将所述目标图像输入训练完成的息肉分割模型，获得所述目标图像对应的息肉分割结果，其中，所述息肉分割模型为基于权利要求1-8中任一所述的息肉分割模型的训练方法进行训练所得的。
12.一种计算机可读介质，其上存储有计算机程序，其特征在于，该程序被处理装置执行时实现权利要求1-9中任一项所述方法的步骤。
13.一种电子设备，其特征在于，包括：
存储装置，其上存储有计算机程序；
处理装置，用于执行所述存储装置中的所述计算机程序，以实现权利要求1-9中任一项所述方法的步骤。
CN202211015296.5A 2022-08-23 2022-08-23 息肉分割模型的训练方法、分割方法、装置、介质及设备 Pending CN115375656A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202211015296.5A CN115375656A (zh) 2022-08-23 2022-08-23 息肉分割模型的训练方法、分割方法、装置、介质及设备

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202211015296.5A CN115375656A (zh) 2022-08-23 2022-08-23 息肉分割模型的训练方法、分割方法、装置、介质及设备

Publications (1)

Publication Number Publication Date
CN115375656A true CN115375656A (zh) 2022-11-22

Family

ID=84066807

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202211015296.5A Pending CN115375656A (zh) 2022-08-23 2022-08-23 息肉分割模型的训练方法、分割方法、装置、介质及设备

Country Status (1)

Country Link
CN (1) CN115375656A (zh)

Cited By (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116168053A (zh) * 2023-02-28 2023-05-26 抖音视界有限公司 息肉分割模型的训练方法、息肉分割方法及相关装置

Cited By (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116168053A (zh) * 2023-02-28 2023-05-26 抖音视界有限公司 息肉分割模型的训练方法、息肉分割方法及相关装置
CN116168053B (zh) * 2023-02-28 2024-02-02 抖音视界有限公司 息肉分割模型的训练方法、息肉分割方法及相关装置

Similar Documents

Publication Publication Date Title
US11392792B2 (en) Method and apparatus for generating vehicle damage information
CN108427939B (zh) 模型生成方法和装置
CN113313064B (zh) 字符识别方法、装置、可读介质及电子设备
CN110399847B (zh) 关键帧提取方法、装置及电子设备
CN113222983A (zh) 图像处理方法、装置、可读介质和电子设备
CN112766284B (zh) 图像识别方法和装置、存储介质和电子设备
CN113140012B (zh) 图像处理方法、装置、介质及电子设备
CN112598673A (zh) 全景分割方法、装置、电子设备和计算机可读介质
CN115578570A (zh) 图像处理方法、装置、可读介质及电子设备
CN118097157B (zh) 基于模糊聚类算法的图像分割方法及系统
CN112418249A (zh) 掩膜图像生成方法、装置、电子设备和计算机可读介质
CN112861940A (zh) 双目视差估计方法、模型训练方法以及相关设备
CN112800276A (zh) 视频封面确定方法、装置、介质及设备
CN114863124A (zh) 模型训练方法、息肉检测方法、相应装置、介质及设备
CN115375656A (zh) 息肉分割模型的训练方法、分割方法、装置、介质及设备
CN115375657A (zh) 息肉检测模型的训练方法、检测方法、装置、介质及设备
CN115375655A (zh) 息肉检测模型的训练方法、检测方法、装置、介质及设备
CN110852242A (zh) 基于多尺度网络的水印识别方法、装置、设备及存储介质
CN115984868A (zh) 文本处理方法、装置、介质及设备
CN111737575B (zh) 内容分发方法、装置、可读介质及电子设备
CN116704593A (zh) 预测模型训练方法、装置、电子设备和计算机可读介质
CN115269978A (zh) 一种视频标签生成方法、装置、设备及介质
CN115115836A (zh) 图像识别方法、装置、存储介质以及电子设备
CN111797931B (zh) 图像处理方法、图像处理网络训练方法、装置、设备
CN115223113B (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