opencv/samples/cpp/bgfg_segm.cpp

98 lines
2.5 KiB
C++
Raw Normal View History

2011-08-10 10:56:27 +02:00
#include "opencv2/core/core.hpp"
2012-04-30 16:33:52 +02:00
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/video/background_segm.hpp"
#include "opencv2/highgui/highgui.hpp"
2010-11-29 15:00:49 +01:00
#include <stdio.h>
2011-08-10 10:56:27 +02:00
using namespace std;
using namespace cv;
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"
2011-08-10 10:56:27 +02:00
"Usage: \n"
" ./bgfg_segm [--camera]=<use camera, if this key is present>, [--file_name]=<path to movie file> \n\n");
2010-11-30 02:26:29 +01:00
}
2010-11-29 15:00:49 +01:00
2011-08-10 10:56:27 +02:00
const char* keys =
{
2012-04-30 16:33:52 +02:00
"{c |camera |true | use camera or not}"
2011-08-10 10:56:27 +02:00
"{fn|file_name|tree.avi | movie file }"
};
2010-11-29 15:00:49 +01:00
//this is a sample for foreground detection functions
2011-08-10 10:56:27 +02:00
int main(int argc, const char** argv)
2010-11-29 15:00:49 +01:00
{
2011-08-10 10:56:27 +02:00
help();
CommandLineParser parser(argc, argv, keys);
bool useCamera = parser.get<bool>("camera");
string file = parser.get<string>("file_name");
VideoCapture cap;
2010-11-29 15:00:49 +01:00
bool update_bg_model = true;
2011-08-10 10:56:27 +02:00
if( useCamera )
cap.open(0);
2010-11-29 15:00:49 +01:00
else
2011-08-10 10:56:27 +02:00
cap.open(file.c_str());
parser.printParams();
if( !cap.isOpened() )
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
2012-04-30 16:33:52 +02:00
BackgroundSubtractorMOG2 bg_model;//(100, 3, 0.3, 5);
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
2012-04-30 16:33:52 +02:00
//cvtColor(_img, img, COLOR_BGR2GRAY);
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;
}