OpenCV的视频背景/前景分割(背景建模/前景提取)类cv::bgsegm::BackgroundSubtractorGSOC的使用示例代码及运行效果

Posted 2022-06-09 昊虹图像算法

关于类cv::bgsegm::BackgroundSubtractorCNT的大概情况，我已在博文https://blog.csdn.net/wenhao_ir/article/details/125003752进行了介绍，这里就不多说了。

需要注意的是这是OpenCV4新增的类，OpenCV3里是没有的。

下面介绍其成员函数。

继承于基类cv::BackgroundSubtractor的成员函数apply()和getBackgroundImage()的详细介绍见博文：https://blog.csdn.net/wenhao_ir/article/details/125007017

其特有的成员函数官方文档里没有说明，也不知道有还是没有。

我们从其对象的初始化函数可以看出它有如下参数可以设置：

mc: Any = None,
nSamples: Any = None,
replaceRate: Any = None,
propagationRate: Any = None,
hitsThreshold: Any = None,
alpha: Any = None,
beta: Any = None,
blinkingSupressionDecay: Any = None,
blinkingSupressionMultiplier: Any = None,
noiseRemovalThresholdFacBG: Any = None,
noiseRemovalThresholdFacFG: Any = None

各参数的具体的意义大家去查阅谷歌编程之夏的相关比赛资料吧。

接下来给出示例代码，
由于博主用的的OpenCV的C++版本为OpenCV3，而这个类是OpenCV4才有的，所以给大家Python代码。
示例代码如下：
示例代码中用到的视频下载链接：https://pan.baidu.com/s/1X08cwwSE4DUvzT0XvHvpvw?pwd=9yyq

# 博主微信/QQ 2487872782
# 有问题可以联系博主交流
# 有图像处理需求也可联系博主
# 图像处理技术交流QQ群 271891601

# !/usr/bin/env python
# -*- coding: utf-8 -*-
# OpenCV的版本为4.1

import cv2 as cv

# 构造VideoCapture对象
cap = cv.VideoCapture('F:/material/videos/car1.avi')

# 创建一个背景分割器
pBackgroundGSOC = cv.bgsegm.createBackgroundSubtractorGSOC()

while True:
    ret, frame = cap.read()  # 读取视频
    # 当所有帧读取完毕后退出循环
    if ret is False:
        print('视频读取失败或者视频已读取完毕')
        break
    else:
        frame = cv.resize(frame, (0, 0), fx=0.3, fy=0.3)
        FGMask = pBackgroundGSOC.apply(frame)  # 背景分割，并得到前景图像
        BGimgae = pBackgroundGSOC.getBackgroundImage()  # 得到背景图像

        current_frame = int(cap.get(cv.CAP_PROP_POS_FRAMES))
        cv.putText(frame, "Current frame:" + str(current_frame), (20, 40), cv.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255))

        cv.imshow('frame', frame)  # 显示原始帧
        cv.imshow('FGMask', FGMask)  # 显示得到的前景图像
        cv.imshow('BGimgae', BGimgae)  # 显示得到的背景图像

    if cv.waitKey(10) & 0xff == 27:  # 按ESC键退出
        break

cap.release()
cv.destroyAllWindows()

运行结果如下图所示：

为了方便大家观察结果，博主录了个视频供大家参考，视频在线观看和下载链接如下：
链接：https://pan.baidu.com/s/1AKHCoIei9LQA0A-xfLJO8w?pwd=nfpq

延伸阅读：
OpenCV3.0中有哪些视频背景/前景分割(背景建模/前景提取)算法的类,它们各自的算法原理、特点是什么,并附示例代码
OpenCV4中有哪些视频背景/前景分割(背景建模/前景提取)算法的类,它们各自的算法原理、特点是什么,并附示例代码