opencv--背景建模（BSM）

最新推荐文章于 2025-04-25 20:37:11 发布

原创最新推荐文章于 2025-04-25 20:37:11 发布 · 471 阅读

3 ·

CC 4.0 BY-SA版权

opencv 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种在背景不变场景下进行图像分割的方法，利用OpenCV库中的GMM（高斯混合模型）和KNN（K个最近邻）算法，通过实例代码展示了如何从视频中实时提取前景目标。

常见算法：1、图像分割（GMM - 高斯混合模型），2、机器学习（KNN - K个最近邻）

场景：主要运用在背景不变的场景下

代码：

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

using namespace cv;
using namespace std;

int main(int argc, char** argv)
{
	VideoCapture capture;
	capture.open("./video/video_003.avi");
	//VideoCapture capture(0);
	if(!capture.isOpened())
	{
		printf("[%s][%d]could not load video data...\n",__FUNCTION__,__LINE__);
		return -1;
	}
	capture.set(CAP_PROP_FRAME_WIDTH, 320);
	capture.set(CAP_PROP_FRAME_HEIGHT ,240);
	double fps = capture.get(CAP_PROP_FPS);
	printf("[%s][%d]fps = %f\n",__FUNCTION__,__LINE__,fps);
	Size size = Size(capture.get(CAP_PROP_FRAME_WIDTH),capture.get(CAP_PROP_FRAME_HEIGHT));
  	cout << size << endl;
	Mat frame;

	Mat bsmaskMOG2;
	Ptr<BackgroundSubtractor> pMOG2 = createBackgroundSubtractorMOG2();

	Mat bsmaskKNN;
	Ptr<BackgroundSubtractor> pKNN = createBackgroundSubtractorKNN();
	
	Mat kernel = getStructuringElement(MORPH_RECT,Size(3,3),Point(-1,-1));

	while(capture.read(frame))
	{
		imshow("video-demo",frame);
		moveWindow("video-demo",0,0);

		pMOG2->apply(frame,bsmaskMOG2);
		morphologyEx(bsmaskMOG2,bsmaskMOG2,MORPH_OPEN,kernel,Point(-1,-1));
		imshow("MOG2",bsmaskMOG2);
		moveWindow("MOG2",size.width+70,0);

		pKNN->apply(frame,bsmaskKNN);
		morphologyEx(bsmaskKNN,bsmaskKNN,MORPH_OPEN,kernel,Point(-1,-1));
		imshow("KNN",bsmaskKNN);
		moveWindow("KNN",size.width*2+70,0);

		if(waitKey(33) == 27)
		{
			break;
		}
	}
	capture.release();
	waitKey(0);
	return 0;
}

效果：