[发明专利]分割网络的训练及目标分割方法、相关设备

说明书

本申请公开了一种分割网络的训练方法、目标分割方法、电子设备及计算机可读存储介质。该训练方法包括：利用分割网络对包含至少一个目标的训练图像处理，得到主要目标的第一关节点拓扑图，主要目标是至少一个目标中的目标；利用分割网络对第一关节点拓扑图和公共特征图融合，得到融合特征图；利用分割网络对融合特征图进行目标分割，得到第一目标分割结果；基于第一目标分割结果获取第一损失；至少基于第一损失调整分割网络的参数。通过上述方式，能够提高后续分割网络对图像中主要目标的分割准确度。

技术领域

本申请涉及图像处理领域，特别是涉及一种分割网络的训练方法、目标分割方法、电子设备及计算机可读存储介质。

背景技术

目标分割技术可应用于行为分析，步态识别等场景。目标分割技术用于分割出单张图像或视频中的目标区域和背景区域。现有的目标分割方法为，提取图像的特征，对图像的特征进行目标分割，得到目标所在的区域。在图像中仅存在单个目标的情况下，现有的目标分割方法的分割效果良好。但是在图像中存在多个目标的情况下，图像中可能会出现目标被遮挡的情况，进而图像中主要目标可能被遮挡，影响对主要目标的分割效果，进而影响后续的行为分析，步态识别等进程。

发明内容

本申请提供一种分割网络的训练方法、目标分割方法、电子设备及计算机可读存储介质，能够提高对图像中主要目标分割的效果。

为解决上述技术问题，本申请采用的一个技术方案是：提供一种分割网络的训练方法。该训练方法包括：利用分割网络对包含至少一个目标的训练图像处理，得到主要目标的第一关节点拓扑图，主要目标是至少一个目标中的目标；利用分割网络对第一关节点拓扑图和公共特征图融合，得到融合特征图；利用分割网络对融合特征图进行目标分割，得到第一目标分割结果；基于第一目标分割结果获取第一损失；至少基于第一损失调整分割网络的参数。

为解决上述技术问题，本申请采用的另一个技术方案是：提供一种目标分割方法，该方法包括：获取目标图像；将目标图像输入分割网络，得到目标图像中主要目标的分割结果；其中，目标分割网络是基于前述训练方法训练得到的。

为解决上述技术问题，本申请采用的再一个技术方案是：提供一种电子设备，该电子设备包括处理器、与处理器连接的存储器，其中，存储器存储有程序指令；处理器用于执行存储器存储的程序指令以实现上述方法。

为解决上述技术问题，本申请采用的又一个技术方案是：提供一种计算机可读存储介质，存储有程序指令，该程序指令被执行时能够实现上述方法。

通过上述方式，本申请分割网络没有直接对公共特征图进行目标分割，而是将其与主要目标的第一关节点特征图融合得到融合特征图之后，再对融合特征图进行目标分割。由于融合过程增强了公共特征图中的主要目标(关节点特征图)对应的部分的语义信息，因此，最终得到的关于主要目标的第一目标分割结果更加准确。进而，基于本申请训练方法训练得到的分割网络，在后续使用过程中，即使目标图像中有多个目标且存在被遮挡的情况，分割网络也能够准确地分割出主要目标。

附图说明

图1是本申请分割网络的训练方法实施例一的流程示意图；

图2是第一分割结果的示意图；

图3是本申请分割网络的训练方法实施例二的流程示意图；

图4是本申请第一关键点热力图的示意图；

图5是其中一种第二关节点拓扑图的示意图；

图6是另外一种第二关节点拓扑图的示意图；

图7是图3中是113的具体流程示意图；

图8是第一关节点拓扑图的示意图；

图9是本申请分割网络一结构示意图；

图10是本申请分割网络的训练方法实施例三的流程示意图；