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

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

关于类cv::bgsegm::BackgroundSubtractorLSBP的大概算法原理，我已在博文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,
LSBPRadius: Any = None,
TLower: Any = None,
Tupper: Any = None,
Tinc: Any = None,
Tdec: Any = None,
Rscale: Any = None,
Rincdec: Any = None,
noiseRemovaLThresholdFacBG: Any = None,
noiseRemovalThresholdFacFG: Any = None,
LSBPthreshold: Any = None,
minCount: Any = None) -> None

具体各参数的意义大家只有去仔细阅读算法原理的论文才能知道了，我在博文https://blog.csdn.net/wenhao_ir/article/details/125003752介绍了其算法原理论文的标题和摘要。

下面上示例代码：
由于博主用的的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')

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

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 = pBackgroundLSBP.apply(frame)  # 背景分割，并得到前景图像
        BGimgae = pBackgroundLSBP.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()

从这个运行结果来看，其初始参数对于这个测试视频是不适合的，需要调参。这里博主就暂时不去研究怎么去调它的参数了。

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