[发明专利]舱段图像实例分割的方法、装置、电子设备及存储介质

权利要求书

1.一种舱段图像实例分割的方法，其特征在于，包括：

获取待装配舱段图像；所述待装配舱段图像包括第一RGB图像以及第一深度图像；所述第一RGB图像与所述第一深度图像一一对应；

将所述待装配舱段图像输入至训练好的图像实例分割模型，得到所述待装配舱段图像中不同舱段对应区域的提取结果；

其中，所述训练好的图像实例分割模型为利用不同舱段图像以及不同舱段图像对应的真实掩膜图像进行训练后得到。

2.根据权利要求1所述的舱段图像实例分割的方法，其特征在于，在将所述待装配舱段图像输入至训练好的图像实例分割模型，得到所述待装配舱段图像中不同舱段对应区域的提取结果之前，还包括：

获取舱段训练图像以及真实掩膜图像；所述舱段训练图像包括第二RGB图像以及第二深度图像；

采用残差网络对所述舱段训练图像进行图像特征提取，得到特征图；

采用区域建议网络生成所述特征图的建议区域；

采用感兴趣区域对齐网络将所述建议区域映射至所述特征图中；

对所述特征图的建议区域进行分类，得到第一损失；

对所述建议区域进行边界框回归，得到第二损失；

根据所述建议区域生成预测掩膜图像；

根据所述真实掩膜图像以及所述预测掩膜图像确定第三损失；

根据所述第一损失、所述第二损失、所述第三损失确定损失函数；

根据所述损失函数更新所述图像实例分割模型的参数，得到训练好的图像实例分割模型。

3.根据权利要求2所述的舱段图像实例分割的方法，其特征在于，所述获取真实掩膜图像，包括：

提取所述舱段训练图像中各舱段外轮廓，得到图像标注信息；

根据所述图像标注信息确定所述真实掩膜图像。

4.根据权利要求2所述的舱段图像实例分割的方法，其特征在于，在所述采用残差网络对所述舱段训练图像进行图像特征提取之前，还包括：

获取所述舱段训练图像对应的标签信息；

对所述第二深度图像进行格式转换；

对所述舱段训练图像进行扩充，得到扩充后的舱段训练图像；

根据所述舱段训练图像对应的标签信息生成所述扩充后的舱段训练图像的标签信息。

5.根据权利要求2所述的舱段图像实例分割的方法，其特征在于，在所述采用区域建议网络生成所述特征图的建议区域之前，还包括：

对所述特征图构造特征金字塔，得到新的特征图。

6.根据权利要求2所述的舱段图像实例分割的方法，其特征在于，对所述特征图的建议区域进行分类得到第一损失，包括：

获取所述舱段训练图像对应的标签信息；

对所述特征图的建议区域进行分类得到预测值；

根据所述舱段训练图像对应的标签信息、所述预测值确定第一损失。

7.一种舱段图像实例分割的装置，其特征在于，包括：

获取模块，用于获取待装配舱段图像；所述待装配舱段图像包括第一RGB图像以及第一深度图像；所述第一RGB图像与所述第一深度图像一一对应；

处理模块，用于将所述待装配舱段图像输入至训练好的图像实例分割模型，得到所述待装配舱段图像中不同舱段对应区域的提取结果；其中，所述训练好的图像实例分割模型为利用不同舱段图像以及不同舱段图像对应的真实掩膜图像进行训练后得到。

8.一种计算机程序产品，包括计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述方法的步骤。

9.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至6任一项所述方法的步骤。

10.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述方法的步骤。