[发明专利]目标分割的方法、装置及电子设备

说明书

本公开公开了一种目标分割的方法、装置及电子设备，涉及人工智能领域，具体涉及计算机视觉和深度学习技术，具体可用于智慧城市和智能交通场景下。具体实现方案为：对待检测图像进行特征提取，生成第一下采样特征图、第二下采样特征图和第三下采样特征图，生成第三上采样特征图和生成第二上采样特征图；根据所述第二上采样特征图和所述第一下采样特征图生成第一上采样特征图；对所述第一上采样特征图进行解码，以生成目标分割图像。本公开实施例可以根据待检测图像检测其中的目标。通过融合不同尺度的特征图，获取的信息更加丰富，通过光流来对其不同尺度图像的特征，获得的目标分割图像更接近实际情况，目标分割的结果更准确。

技术领域

本公开涉及人工智能领域，具体涉及计算机视觉和深度学习技术，具体可用于智慧城市和智能交通场景下，尤其涉及一种目标分割的方法、装置及电子设备。

背景技术

随着人工智能相关技术的发展与应用，越来越多的领域显露出对智能化、自动化技术的强烈需求，其中短视频领域就是其中之一。在短视频领域中，视频目标分割方法的使用前景非常好，无论是视频中指定目标的抠除，背景虚化等都是比较依赖于视频目标分割方法的。所以视频目标分割方法的发展对短视频处理的智能化，特效处理等具有十分重要的意义。

但是现在视频目标分割方法中检测目标物体的准确度较低，目前缺乏可以较为精确地检测目标物体的视频目标分割方法。

发明内容

本公开提供了一种目标分割的方法、装置及电子设备，可用于智慧城市和智能交通场景下。

根据本公开的第一方面，提供了一种目标分割的方法，包括：

对待检测图像进行特征提取，生成第一下采样特征图、第二下采样特征图和第三下采样特征图；

根据所述第三下采样特征图生成第三上采样特征图；

根据所述第三上采样特征图和所述第二下采样特征图生成第二上采样特征图；

根据所述第二上采样特征图和所述第一下采样特征图生成第一上采样特征图；

对所述第一上采样特征图进行解码，以生成目标分割图像。

可选地，所述生成第一下采样特征图、第二下采样特征图和第三下采样特征图，包括：

提取所述待检测图像的特征，以生成所述第一下采样特征图；

对所述第一下采样特征图进行下采样，以生成所述第二下采样特征图；

对所述第二下采样特征图进行下采样，以生成所述第三下采样特征图。

可选地，所述根据所述第三下采样特征图生成第三上采样特征图，包括：

对所述第三下采样特征图进行池化，以生成所述第三上采样特征图。

可选地，所述根据所述第三上采样特征图和所述第二下采样特征图生成第二上采样特征图，包括：

对所述第三上采样特征图进行上采样，以生成第一特征图；

将所述第一特征图与所述第二下采样特征图拼接，以生成第一融合特征图；

将所述第一融合特征图输入光流提取网络，并生成第一光流特征图；

通过所述第一光流特征图修正所述第一特征图中像素的位置，以生成第一修正特征图；

将所述第一修正特征图中像素与所述第二下采样特征图中像素点对点相加，以生成所述第二上采样特征图。

可选地，所述根据所述第二上采样特征图和所述第一下采样特征图生成第一上采样特征图，包括：

对所述第二上采样特征图进行上采样，以生成第二特征图；