[发明专利]半监督视频目标分割方法及装置

说明书

本发明公开了一种半监督视频目标分割方法及装置，该方法包括：获取待分割视频的模板帧图像、模板帧图像分割标签及待查询帧图像；将模板帧图像及模板帧图像分割标签输入至第一编码器提取特征，得到复合特征；将待查询帧图像输入至第二编码器提取特征，得到图像特征；将复合特征及图像特征输入至预先训练的时序推断模块进行分割结果预测，得到待查询帧图像对应的预测分割结果。本发明提供的方案，基于预测分割结果的循环一致性进行半监督训练，在仅需要一段视频中第一帧图像及第一帧图像分割标签的前提下，能高效地进行训练，基于此所进行的视频目标分割仍然具有较高的预测精度，从而无需人工一一进行标注，节省了时间成本，同时提高了分割效率。

技术领域

本发明涉及计算机技术领域，具体涉及一种半监督视频目标分割方法及装置。

背景技术

视频目标自动分割是视频处理技术中的重要任务，已经在影像留色，视频会议和自动驾驶等领域取得了广泛的应用。已有的视频目标分割方法使用完整的视频目标分割标签来进行训练，需要耗费大量的人力和时间对视频中的每一帧进行像素级标注。

发明内容

鉴于上述问题，提出了本发明以便提供一种克服上述问题或者至少部分地解决上述问题的半监督视频目标分割方法及装置。

根据本发明的一个方面，提供了一种半监督视频目标分割方法，包括：

获取待分割视频的模板帧图像、模板帧图像分割标签及待查询帧图像；

将模板帧图像及模板帧图像分割标签输入至第一编码器提取特征，得到复合特征；

将待查询帧图像输入至第二编码器提取特征，得到图像特征；

将复合特征及图像特征输入至预先训练的时序推断模块进行分割结果预测，得到待查询帧图像对应的预测分割结果。

根据本发明的另一方面，提供了一种半监督视频目标分割装置，包括：

获取模块，适于获取待分割视频的模板帧图像、模板帧图像分割标签及待查询帧图像；

特征提取模块，适于将模板帧图像及模板帧图像分割标签输入至第一编码器提取特征，得到复合特征；及，将待查询帧图像输入至第二编码器提取特征，得到图像特征；

预测模块，适于将复合特征及图像特征输入至预先训练的时序推断模块进行分割结果预测，得到待查询帧图像对应的预测分割结果。

根据本发明的又一方面，提供了一种计算设备，包括：处理器、存储器、通信接口和通信总线，处理器、存储器和通信接口通过通信总线完成相互间的通信；

存储器用于存放至少一可执行指令，可执行指令使处理器执行上述半监督视频目标分割方法对应的操作。

根据本发明的再一方面，提供了一种计算机存储介质，存储介质中存储有至少一可执行指令，可执行指令使处理器执行如上述半监督视频目标分割方法对应的操作。

本发明提供的方案，基于预测分割结果的循环一致性进行半监督训练，在仅需要一段视频中第一帧图像及第一帧图像分割标签的前提下，能高效地进行训练，基于此所进行的视频目标分割仍然具有较高的预测精度，从而无需人工一一进行标注，节省了时间成本，同时提高了分割效率。

上述说明仅是本发明技术方案的概述，为了能够更清楚了解本发明的技术手段，而可依照说明书的内容予以实施，并且为了让本发明的上述和其它目的、特征和优点能够更明显易懂，以下特举本发明的具体实施方式。

附图说明

通过阅读下文优选实施方式的详细描述，各种其他的优点和益处对于本领域普通技术人员将变得清楚明了。附图仅用于示出优选实施方式的目的，而并不认为是对本发明的限制。而且在整个附图中，用相同的参考符号表示相同的部件。在附图中：