[发明专利]一种基于目标检测的钻探深度统计方法及系统

权利要求书

1.一种基于目标检测的钻探深度统计方法，其特征在于，包括：

将钻机的图像帧输入预先训练的目标检测模型，识别所述图像帧中钻机的堵头和钻杆尾部的坐标，得到第一识别结果；

根据所述第一识别结果计算堵头和钻杆尾部的像素距离；

若所述图像帧与上一帧图像中堵头和钻杆尾部的像素距离之差大于预设阈值，则钻杆数量加1；

根据已知的钻杆长度和所述钻杆数量计算钻探深度。

2.根据权利要求1所述的方法，其特征在于，在根据第一识别结果确定堵头和钻杆尾部的像素距离之前，还包括：

计算所述图像帧之前预设数量的图像帧中钻机堵头坐标的平均值；

根据第一识别结果确定所述堵头和钻杆尾部的像素距离，具体包括：

根据所述钻机堵头坐标的平均值和钻杆尾部的坐标计算堵头和钻杆尾部的像素距离。

3.根据权利要求1或2所述的方法，其特征在于，在根据第一识别结果计算堵头和钻杆尾部的像素距离之前，还包括：

识别所述图像帧以及之前相隔预设帧数的图像帧中钻机的钻杆尾部和主机的坐标，得到第二识别结果；

根据所述第二识别结果判断所述第一识别结果中的钻杆尾部的坐标是否有效；

若有效，则执行所述根据第一识别结果计算堵头和钻杆尾部的像素距离的步骤，否则不执行。

4.根据权利要求3所述的方法，其特征在于，所述根据所述第二识别结果判断所述第一识别结果中的钻杆尾部的坐标是否有效，具体包括：

根据所述图像帧以及之前相隔预设帧数的图像帧中钻机的钻杆尾部和主机的坐标，计算钻杆尾部和主机的运动向量；

根据钻杆尾部和主机的运动向量的夹角，判断所述第一识别结果中的钻杆尾部的坐标是否有效。

5.一种基于目标检测的钻探深度统计系统，其特征在于，包括：

对象识别模块，用于将钻机的图像帧输入预先训练的目标检测模型，识别所述图像帧中钻机的堵头和钻杆尾部的坐标，得到第一识别结果；

距离计算模块，用于根据所述第一识别结果计算堵头和钻杆尾部的像素距离；

钻杆技术模块，用于若所述图像帧与上一帧图像中堵头和钻杆尾部的像素距离之差大于预设阈值，则钻杆数量加1；

深度计算模块，用于根据已知的钻杆长度和所述钻杆数量计算钻探深度。

6.根据权利要求5所述的系统，其特征在于，还包括：

堵头坐标计算模块，用于在所述距离计算模块根据第一识别结果计算堵头和钻杆尾部的像素距离之前，计算所述图像帧之前预设数量的图像帧中钻机堵头坐标的平均值；

所述距离计算模块，具体用于：

根据所述钻机堵头坐标的平均值和钻杆尾部的坐标计算堵头和钻杆尾部的像素距离。

7.根据权利要求5或6所述的系统，其特征在于，所述对象识别模块，还用于：

识别所述图像帧以及之前相隔预设帧数的图像帧中钻机的钻杆尾部和主机的坐标，得到第二识别结果；

所述系统还包括：

钻杆尾部坐标筛选模块，用于根据所述第二识别结果判断所述第一识别结果中的钻杆尾部的坐标是否有效；

所述距离计算模块，用于若所述钻杆尾部坐标筛选模块判断所述第一识别结果中的钻杆尾部的坐标有效，则根据第一识别结果计算堵头和钻杆尾部的像素距离。

8.根据权利要求7所述的系统，其特征在于，所述钻杆尾部坐标筛选模块，具体包括：

运动向量计算单元，用于根据所述图像帧以及之前相隔预设帧数的图像帧中钻机的钻杆尾部和主机的坐标，计算钻杆尾部和主机的运动向量；

夹角判断单元，用于根据钻杆尾部和主机的运动向量的夹角，判断所述第一识别结果中的钻杆尾部的坐标是否有效。

9.一种终端设备，其特征在于，包括：

处理器；以及

存储器，其上存储有可执行代码，当所述可执行代码被所述处理器执行时，使所述处理器执行如权利要求1-4中任一项所述的方法。

10.一种非暂时性机器可读存储介质，其上存储有可执行代码，其特征在于，当所述可执行代码被电子设备的处理器执行时，使所述处理器执行如权利要求1-4中任一项所述的方法。