[发明专利]道路提取方法、装置、设备以及存储介质

权利要求书

1.一种道路提取方法，包括：

获取遥感影像；

利用预先训练的第一斜框标注模型对所述遥感影像进行斜框标注，得到所述遥感影像的道路的斜框标注信息；

基于所述遥感影像的道路的斜框标注信息，提取所述遥感影像的道路信息；

其中，所述利用预先训练的第一斜框标注模型对所述遥感影像进行斜框标注，得到所述遥感影像的道路的斜框标注信息，包括：

将所述遥感影像输入至所述第一斜框标注模型，得到所述遥感影像的道路的斜框标注信息；

其中，所述斜框标注模型的训练步骤包括：

获取样本道路分割标注图片，其中，所述样本道路分割标注图片的道路标注有斜框标注信息；

将所述样本道路分割标注图片作为输入，将所述样本道路分割标注图片的道路的斜框标注信息作为监督，训练得到第二斜框标注模型；

获取样本遥感影像；

对所述样本遥感影像进行道路标注，得到样本标注遥感影像；

将所述样本标注遥感影像输入至所述第二斜框标注模型，得到所述样本遥感影像的道路的斜框标注信息；

将所述样本遥感影像作为输入，将所述样本遥感影像的道路的斜框标注信息作为监督，训练得到所述第一斜框标注模型。

2.根据权利要求1所述的方法，其中，所述利用预先训练的第一斜框标注模型对所述遥感影像进行斜框标注，得到所述遥感影像的道路的斜框标注信息，还包括：

对所述遥感影像进行道路标注，得到标注遥感影像；

将所述标注遥感影像输入至所述第一斜框标注模型，得到所述遥感影像的道路的斜框标注信息。

3.根据权利要求2所述的方法，其中，所述第一斜框标注模型的训练步骤还包括：

获取样本道路分割标注图片，其中，所述样本道路分割标注图片的道路标注有斜框标注信息；

将所述样本道路分割标注图片作为输入，将所述样本道路分割标注图片的道路的斜框标注信息作为监督，训练得到所述第一斜框标注模型。

4.根据权利要求1或3所述的方法，其中，所述样本道路分割标注图片通过如下步骤生成：

生成标签图片中的道路数量和道路宽度；

对于所述标签图片中的每条道路，生成所述道路的起点和终点；

在所述起点和所述终点之间生成至少一个点；

对于相邻的两个点，由上一点至下一点绘制线段，其中，所述线段的宽度等于所述道路宽度；

基于上一点和下一点的坐标，生成斜框标注信息，其中，所述斜框标注信息包括斜框的对角线交点、宽、高和倾斜角度。

5.一种道路提取装置，包括：

获取模块，被配置成获取遥感影像；

标注模块，被配置成利用预先训练的第一斜框标注模型对所述遥感影像进行斜框标注，得到所述遥感影像的道路的斜框标注信息；

提取模块，被配置成基于所述遥感影像的道路的斜框标注信息，提取所述遥感影像的道路信息；

其中，所述标注模块进一步被配置成：

将所述遥感影像输入至所述第一斜框标注模型，得到所述遥感影像的道路的斜框标注信息；

其中，所述装置还包括训练模块，所述训练模块被配置成：

获取样本道路分割标注图片，其中，所述样本道路分割标注图片的道路标注有斜框标注信息；

将所述样本道路分割标注图片作为输入，将所述样本道路分割标注图片的道路的斜框标注信息作为监督，训练得到第二斜框标注模型；

获取样本遥感影像；

对所述样本遥感影像进行道路标注，得到样本标注遥感影像；

将所述样本标注遥感影像输入至所述第二斜框标注模型，得到所述样本遥感影像的道路的斜框标注信息；

将所述样本遥感影像作为输入，将所述样本遥感影像的道路的斜框标注信息作为监督，训练得到所述第一斜框标注模型。

6.根据权利要求5所述的装置，其中，所述标注模块还进一步被配置成：

对所述遥感影像进行道路标注，得到标注遥感影像；

将所述标注遥感影像输入至所述第一斜框标注模型，得到所述遥感影像的道路的斜框标注信息。