101 lines
2.5 KiB
C++
101 lines
2.5 KiB
C++
#include "opencv2/core.hpp"
|
|
#include <opencv2/core/utility.hpp>
|
|
#include "opencv2/imgproc.hpp"
|
|
#include "opencv2/video/background_segm.hpp"
|
|
#include "opencv2/videoio.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
#include <stdio.h>
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
|
|
static void help()
|
|
{
|
|
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 [--camera]=<use camera, if this key is present>, [--file_name]=<path to movie file> \n\n");
|
|
}
|
|
|
|
const char* keys =
|
|
{
|
|
"{c camera | | use camera or not}"
|
|
"{fn file_name|tree.avi | movie file }"
|
|
};
|
|
|
|
//this is a sample for foreground detection functions
|
|
int main(int argc, const char** argv)
|
|
{
|
|
help();
|
|
|
|
CommandLineParser parser(argc, argv, keys);
|
|
bool useCamera = parser.has("camera");
|
|
string file = parser.get<string>("file_name");
|
|
VideoCapture cap;
|
|
bool update_bg_model = true;
|
|
|
|
if( useCamera )
|
|
cap.open(0);
|
|
else
|
|
cap.open(file.c_str());
|
|
|
|
parser.printMessage();
|
|
|
|
if( !cap.isOpened() )
|
|
{
|
|
printf("can not open camera or video file\n");
|
|
return -1;
|
|
}
|
|
|
|
namedWindow("image", WINDOW_NORMAL);
|
|
namedWindow("foreground mask", WINDOW_NORMAL);
|
|
namedWindow("foreground image", WINDOW_NORMAL);
|
|
namedWindow("mean background image", WINDOW_NORMAL);
|
|
|
|
Ptr<BackgroundSubtractor> bg_model = createBackgroundSubtractorMOG2();
|
|
|
|
Mat img, fgmask, fgimg;
|
|
|
|
for(;;)
|
|
{
|
|
cap >> img;
|
|
|
|
if( img.empty() )
|
|
break;
|
|
|
|
//cvtColor(_img, img, COLOR_BGR2GRAY);
|
|
|
|
if( fgimg.empty() )
|
|
fgimg.create(img.size(), img.type());
|
|
|
|
//update the model
|
|
bg_model->apply(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);
|
|
if( k == 27 ) break;
|
|
if( k == ' ' )
|
|
{
|
|
update_bg_model = !update_bg_model;
|
|
if(update_bg_model)
|
|
printf("Background update is on\n");
|
|
else
|
|
printf("Background update is off\n");
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|