[发明专利]用于生成塔机吊臂信息的方法、装置、设备和介质

权利要求书

1.一种用于生成塔机吊臂信息的方法，包括：

将目标塔机图像输入到预先训练的塔机粗定位检测模型，得到所述目标塔机图像对应的标注信息，所述标注信息用于表征塔机图像中的塔机预设部分显示范围的范围信息；

根据所述范围信息，对所述目标塔机图像进行截取，得到第一塔机图像；

将所述第一塔机图像输入到预先训练的塔机吊臂部分检测模型，得到所述第一塔机图像对应的第一标注信息，所述第一标注信息用于表征第一塔机图像中的吊臂显示范围中的第一范围信息，其中，所述塔机吊臂部分检测模型是通过以下步骤训练得到：

获取训练样本集合，训练样本包括样本第一塔机图像和与样本第一塔机图像对应的第一标注信息；

将所述训练样本集合中的训练样本的样本第一塔机图像作为输入，将与输入的样本第一塔机图像对应的第一标注信息作为期望输出，训练得到所述塔机吊臂部分检测模型；

根据所述目标塔机图像和所述第一范围信息，得到预定形状图像；

将所述预定形状图像输入到预先训练的塔机吊臂关键点检测模型，得到所述预定形状图像中的关键点信息；

基于所述第一范围信息和关键点信息，生成塔机吊臂信息。

2.根据权利要求1所述的方法，其中，所述根据所述目标塔机图像和所述第一范围信息，得到预定形状图像，包括：

将第一范围向所述第一塔机图像中显示的吊臂所在方向延伸至所述目标塔机图像的边界，生成第一区域；

将所述第一区域进行像素填充处理，得到预定形状图像。

3.根据权利要求1所述的方法，其中，所述关键点信息包括第一关键点信息和第二关键点信息；以及

基于所述第一范围信息和关键点信息，生成塔机吊臂信息，包括：

根据所述第一范围信息，第一关键信息，第二关键信息，确定塔机吊臂信息。

4.根据权利要求1-3之一所述的方法，其中，所述塔机吊臂关键点检测模型是通过以下步骤训练得到：

获取训练样本集合，训练样本包括样本预定形状图像和样本预定形状图像中的关键点信息；

将所述训练样本集合中的训练样本的样本预定形状图像作为输入，将与输入的样本预定形状图像对应的样本预定形状图像中的关键点信息作为期望输出，训练得到所述塔机吊臂关键点检测模型。

5.根据权利要求1所述的方法，其中，所述方法还包括：

将所述塔机吊臂信息发送到控制终端，其中，控制终端：根据所述塔机吊臂信息，控制塔机进行目标操作。

6.一种电子设备，包括：

一个或多个处理器；

存储装置，其上存储有一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-5中任一所述的方法。

7.一种计算机可读介质，其上存储有计算机程序，其中，所述程序被处理器执行时实现如权利要求1-5中任一所述的方法。