Added displaying of the mean background image in the bgfg_segm sample (ticket #317).

This commit is contained in:
Ilya Lysenkov 2011-06-03 14:13:03 +00:00
parent 04ebfc0a02
commit 5d4afe81e0

View File

@ -31,12 +31,14 @@ int main(int argc, char** argv)
return -1;
}
namedWindow("BG", 1);
namedWindow("FG", 1);
namedWindow("image", CV_WINDOW_NORMAL);
namedWindow("foreground mask", CV_WINDOW_NORMAL);
namedWindow("foreground image", CV_WINDOW_NORMAL);
namedWindow("mean background image", CV_WINDOW_NORMAL);
BackgroundSubtractorMOG2 bg_model;
Mat img, fgmask;
Mat img, fgmask, fgimg;
for(;;)
{
cap >> img;
@ -44,10 +46,24 @@ int main(int argc, char** argv)
if( img.empty() )
break;
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);
if( k == 27 ) break;
if( k == ' ' )