Wang's blog

BGSLibrary背景提取库介绍

Published on

简介

BGSLibrary(Background Subtraction Library) 是一个背景减除(背景提取)库,它使用C++开发,提供Python/Matlab/Java接口,并可以跨平台使用(Windows/Ubuntu/OS X)。

安装

可以编译安装,也可以安装二进制包。但是对于python可以直接使用pip安装,非常方便:

pip install pybgs

使用

根据官方demo,使用方法十分简单(以python为例):

import numpy as np
import cv2
import pybgs as bgs

algorithm = bgs.FrameDifference()
video_file = "dataset/video.avi"
capture = cv2.VideoCapture(video_file)
while True:
    flag, frame = capture.read()
    if not flag:
        break

    img_output = algorithm.apply(frame)
    img_bgmodel = algorithm.getBackgroundModel()

    cv2.imshow('video', frame)
    cv2.imshow('img_output', img_output)
    cv2.imshow('img_bgmodel', img_bgmodel)

    if cv2.waitKey(10) == 27:
        break

所有算法具有相同接口,如需更换算法只需修改algorithm变量即可。

支持的算法

支持算法列表