尉来集团-油田装备维修保养与再制造专家

　　步骤1：对给定的抽油机视频，先截取成一帧帧的图片，准备对图片中的抽油机进行目标检测；

　　步骤3：用训练好的yolov5模型对图片中的抽油机进行检测，得到抽油机的boundingbox，包括左上角(x1，y1)、右下角(x2，y2)；

　　步骤4：在当前帧，我们把抽油机的boundingbox区域裁剪出来作为图片2；定位到当前帧前t秒的图片，把该帧对应的同位置区域裁剪出来作为图片1；

　　步骤5：用滑动窗口在两张图片上取m×n的窗口，得到m×n个窗口a和m×n个窗口b，计算每两个对应窗口的相似度；

　　步骤6：分别求出窗口a的均值μa、方差窗口b的均值μb、方差以及窗口a与窗口b的协方差σab；

　　步骤11：对每次滑动窗口计算得到m(a，b)，计算m×n次得到每两个窗口的相似度，对相似度求平均值作为全局的相似度m(x，y)；

　　步骤12：全局相似度m(x，y)的值总处于0到1之间，设定阈值p。当m(x，y)≥p，前后两张图相似度高，判定抽油机停止运行；当m(x，y)≤p，前后两张图相似度低，判定抽油机正在运行。

　　2.根据权利要求1所述的一种视频抽油机启停检测的方法，其特征在于：所述步骤2的数据集为步骤1在视频中截取图片的集合。

　　3.根据权利要求1所述的一种视频抽油机启停检测的方法，其特征在于：步骤4中的当前帧为在图片集中任意一张图片，其中定位当前帧前或后的时间t为正数。

　　4.根据权利要求1所述的一种视频抽油机启停检测的方法，其特征在于：步骤5的图片为步骤4截取的图片1和2，所述1和2截取的窗口a和窗口b大小位置相同。

　　5.根据权利要求1所述的一种视频抽油机启停检测的方法，其特征在于：步骤6中的μa、μb、和为步骤5中的图片在步骤7中表示为图片的亮度数据，步骤6中的μa、μb、和为步骤5中的图片在步骤8中表示为图片的灰阶数据，步骤6中的μa、μb、和为步骤5中的图片在步骤9中表示为图片的汉明距离数据。

　　6.根据权利要求1所述的一种视频抽油机启停检测的方法，其特征在于：步骤10中的总相似度表达为步骤7、8和9中的亮度、对比度和结构相似度的乘积。

　　7.根据权利要求1所述的一种视频抽油机启停检测的方法，其特征在于：步骤12中设定的阈值为0.8。

　　本发明涉及图像目标检测、图像相似度对比技术领域，具体地说，涉及一种视频抽油机启停检测的方法主要包括对给定的抽油机视频，先截取成一帧帧的图片，准备对图片中的抽油机进行目标检测；标注抽油机的数据集，用来训练Yolov5目标检测模型；用训练好的Yolov5模型对图片中的抽油机进行检测，得到抽油机的Bounding Box，包括左上角(x1，y1)、右下角(x2，y2)；在当前帧，我们把抽油机的Bounding Box区域裁剪出来作为图片2；定位到当前帧前或后T秒的图片，把该帧对应的同位置区域裁剪出来作为图片1；用滑动窗口在两张图片上取M×N的窗口，计算每两个对应窗口的相似度，求平均得到两张图片总的相似度。该发明首先从视频中截取出的两帧图片，再对这两帧图片进行目标检测得到目标区域，对两个区域进行对比分析来判断抽油机的启停。