OpenCV - 视频分析与对象跟踪实战（02. 背景消除建模（BSM））

背景消除建模（BSM）

在背景不怎么变化是视频中使用
利用前几帧的图像作为背景模型，后续的帧图像与背景模型比较，得到差异就是前景对象

• 基本原理
  Background Subtraction基本原理（BS算法）

  • 图像分割(GMM –高斯混合模型)
  • 机器学习(KNN –K个最近邻)
    

• 相关API

  • BackgroundSubtractor
    • BackgroundSubtractorMOG2
    • BackgroundSubtractorKNN

• 代码演示

#include<opencv2/opencv.hpp>
#include<iostream>

using namespace cv;
using namespace std;

int main(int argc, char** argv){
	VideoCapture capture;
	capture.open("E:/Experiment/OpenCV/src/video_003.avi");
	if (!capture.isOpened()){
		printf("Video could not load ...");
		return -1;
	}

	Mat frame;
	Mat bsmaskMOG2, bsmaskKNN;
	namedWindow("test-show", CV_WINDOW_AUTOSIZE);
	namedWindow("MOG2-show", CV_WINDOW_AUTOSIZE);
	namedWindow("KNN-show", CV_WINDOW_AUTOSIZE);
	// 去除噪点
	Mat kernel = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1));

	// intialization BS
	Ptr<BackgroundSubtractor> pMOG2 = createBackgroundSubtractorMOG2();
	Ptr<BackgroundSubtractor> pKNN = createBackgroundSubtractorKNN();
	while (capture.read(frame)){
		imshow("test-show", frame);
		
		// MOG BS
		pMOG2->apply(frame, bsmaskMOG2);
		morphologyEx(bsmaskMOG2, bsmaskMOG2, MORPH_OPEN, kernel, Point(-1, -1));//进行开操作，去除噪声
		imshow("MOG2-show", bsmaskMOG2);
		
		// KNN BS
		pKNN->apply(frame, bsmaskKNN);
		morphologyEx(bsmaskKNN, bsmaskKNN, MORPH_OPEN, kernel, Point(-1, -1));//进行开操作，去除噪声
		imshow("KNN-show", bsmaskKNN);

		char c = waitKey(50);
		if (c == 27){
			break;
		}
	}

	capture.release();
	waitKey(0);
	return 0;
}