一种视频识别中线状目标张紧状态的判定方法

专利主权项内容

1.一种视频识别中线状目标张紧状态的判定方法，其特征在于，包括如下步骤：S1：采集包含石油井筒作业场景中待判定线状目标的图像数据，得到原数据集；S2：分别对原数据集中每张图像中的待判定线状目标的外接矩形框进行标注，得到第一训练集；S3：基于卷积神经网络，训练第一训练集，得到用于检测所述待判定线状目标的外接矩形框的目标检测模型；S4：将原数据集以外的包含所述待判定线状目标的图像作为第一目标图像，确定该第一目标图像中待判定线状目标的位置；S5：确定第一目标图像中待判定线状目标的轮廓，具体包括：将原数据集中每张图像数据中的待判定线状目标的轮廓进行标注，得到第二训练集；基于卷积神经网络，训练第二训练集，得到用于检测所述待判定线状目标的轮廓的目标分割模型；裁剪经步骤S4所得的第一目标图像的外接矩形框，得到w*h像素的图像，作为第二目标图像；将第二目标图像输入至所述目标分割模型中，经目标分割模型推理，输出一组分割掩码，一组分割掩码为与第二目标图像具有相同尺寸的二进制图像，一组分割掩码又包括多个分割掩码，其中，每个分割掩码都一一对应于第二目标图像中的每个像素，每个分割掩码表示其对应像素所属的区域；对每个分割掩码进行二值化，其中，分割掩码的值为0，表示其对应的像素属于所述待判定线状目标的轮廓区域外；分割掩码的值为1，表示其对应的像素属于所述待判定线状目标的轮廓区域内；S6：判定第一目标图像中的待判定线状目标是否处于张紧状态，具体包括：以第一目标图像左上角为原点，建立直角坐标系，将经步骤S5所得取值为1的分割掩码对应的像素的点坐标存入列表中，得到第一目标图像中的待判定线状目标轮廓的列表；从列表中选取距离最远的两个点，确定一条直线y=kx+a，其中，x、y分别表示待判定线状目标对应直线的自变量和因变量，k表示待判定线状目标对应直线在坐标系中的斜率，a表示待判定线状目标对应直线在坐标系中的截距；以及，坐标系中，x轴正向向右，y轴正向向下；如果列表中除距离最远的两个点外，其它点均匀分布在直线两侧，则第一目标图像中的待判定线状目标处于张紧状态；反之，如果列表中的其它点无规律地分布，则第一目标图像中的待判定线状目标未处于张紧状态。 该数据由整理