opencv/samples/cpp/bgfg_segm.cpp

89 lines
2.2 KiB
C++
Raw Normal View History

#include "opencv2/core/core.hpp"
#include "opencv2/video/background_segm.hpp"
#include "opencv2/highgui/highgui.hpp"
2010-11-29 15:00:49 +01:00
#include <stdio.h>
using namespace cv;
using namespace std;
2010-11-30 02:26:29 +01:00
void help()
{
2010-12-04 09:30:10 +01:00
printf("\nDo background segmentation, especially demonstrating the use of cvUpdateBGStatModel().\n"
" Learns the background at the start and then segments.\n"
" Learning is togged by the space key. Will read from file or camera\n"
"Usage: \n"
" ./bgfg_segm [--file_name]=<input file, camera as defautl>\n\n");
2010-11-30 02:26:29 +01:00
}
2010-11-29 15:00:49 +01:00
//this is a sample for foreground detection functions
int main(int argc, const char** argv)
2010-11-29 15:00:49 +01:00
{
help();
CommandLineParser parser(argc, argv);
string fileName = parser.get<string>("file_name", "0");
VideoCapture cap;
2010-11-29 15:00:49 +01:00
bool update_bg_model = true;
if(fileName == "0" )
cap.open(0);
2010-11-29 15:00:49 +01:00
else
cap.open(fileName.c_str());
if( !cap.isOpened() )
2010-11-29 15:00:49 +01:00
{
help();
2010-11-29 15:00:49 +01:00
printf("can not open camera or video file\n");
return -1;
}
namedWindow("image", CV_WINDOW_NORMAL);
namedWindow("foreground mask", CV_WINDOW_NORMAL);
2011-06-03 16:13:43 +02:00
namedWindow("foreground image", CV_WINDOW_NORMAL);
namedWindow("mean background image", CV_WINDOW_NORMAL);
2010-11-29 15:00:49 +01:00
BackgroundSubtractorMOG2 bg_model;
Mat img, fgmask, fgimg;
for(;;)
2010-11-29 15:00:49 +01:00
{
cap >> img;
if( img.empty() )
break;
2010-11-29 15:00:49 +01:00
if( fgimg.empty() )
fgimg.create(img.size(), img.type());
//update the model
bg_model(img, fgmask, update_bg_model ? -1 : 0);
fgimg = Scalar::all(0);
img.copyTo(fgimg, fgmask);
Mat bgimg;
bg_model.getBackgroundImage(bgimg);
imshow("image", img);
imshow("foreground mask", fgmask);
imshow("foreground image", fgimg);
if(!bgimg.empty())
imshow("mean background image", bgimg );
char k = (char)waitKey(30);
2010-11-29 15:00:49 +01:00
if( k == 27 ) break;
if( k == ' ' )
2010-11-30 02:26:29 +01:00
{
2010-11-29 15:00:49 +01:00
update_bg_model = !update_bg_model;
2010-11-30 02:26:29 +01:00
if(update_bg_model)
printf("Background update is on\n");
else
printf("Background update is off\n");
}
2010-11-29 15:00:49 +01:00
}
return 0;
}