mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-11 03:33:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #10868 from mshabunin:sample-bgsegm

Browse Source 
* Samples: reworked bgfg_bgsegm

* fixup! Samples: reworked bgfg_bgsegm
This commit is contained in:
Maksim Shabunin 2018-02-17 15:04:14 +03:00 committed by Alexander Alekhin
parent 73a8369631
commit aa7a964139
1 changed files with 86 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

166
samples/cpp/bgfg_segm.cpp
View File

@ -1,111 +1,117 @@
#include "opencv2/core.hpp" #include "opencv2/core.hpp"
#include <opencv2/core/utility.hpp>
#include "opencv2/imgproc.hpp" #include "opencv2/imgproc.hpp"
#include "opencv2/video/background_segm.hpp" #include "opencv2/video.hpp"
#include "opencv2/videoio.hpp" #include "opencv2/videoio.hpp"
#include "opencv2/highgui.hpp" #include "opencv2/highgui.hpp"
#include <stdio.h> #include <iostream>
using namespace std; using namespace std;
using namespace cv; 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}"
"{m method |mog2 | method (knn or mog2) }"
"{s smooth | | smooth the mask }"
"{fn file_name|../data/tree.avi | movie file }"
};
//this is a sample for foreground detection functions
int main(int argc, const char** argv) int main(int argc, const char** argv)
{ {
help(); const String keys = "{c camera||use video stream from camera (default is NO)}"
"{fn file_name|../data/tree.avi|video file}"
"{m method|mog2|method: background subtraction algorithm ('knn', 'mog2')}"
"{h help||show help message}";
CommandLineParser parser(argc, argv, keys); CommandLineParser parser(argc, argv, keys);
parser.about("This sample demonstrates background segmentation.");
if (parser.has("help"))
{
parser.printMessage();
return 0;
}
bool useCamera = parser.has("camera"); bool useCamera = parser.has("camera");
bool smoothMask = parser.has("smooth"); String file = parser.get<String>("file_name");
string file = parser.get<string>("file_name"); String method = parser.get<String>("method");
string method = parser.get<string>("method"); if (!parser.check())
VideoCapture cap; {
bool update_bg_model = true; parser.printErrors();
return 1;
}
if( useCamera ) VideoCapture cap;
if (useCamera)
cap.open(0); cap.open(0);
else else
cap.open(file.c_str()); cap.open(file.c_str());
if (!cap.isOpened())
parser.printMessage();
if( !cap.isOpened() )
{ {
printf("can not open camera or video file\n"); cout << "Can not open video stream: '" << (useCamera ? "<camera 0>" : file) << "'" << endl;
return -1; return 2;
} }
namedWindow("image", WINDOW_NORMAL); Ptr<BackgroundSubtractor> model;
namedWindow("foreground mask", WINDOW_NORMAL); if (method == "knn")
namedWindow("foreground image", WINDOW_NORMAL); model = createBackgroundSubtractorKNN();
namedWindow("mean background image", WINDOW_NORMAL); else if (method == "mog2")
model = createBackgroundSubtractorMOG2();
Ptr<BackgroundSubtractor> bg_model = method == "knn" ? if (!model)
createBackgroundSubtractorKNN().dynamicCast<BackgroundSubtractor>() :
createBackgroundSubtractorMOG2().dynamicCast<BackgroundSubtractor>();
Mat img0, img, fgmask, fgimg;
for(;;)
{ {
cap >> img0; cout << "Can not create background model using provided method: '" << method << "'" << endl;
return 3;
}
if( img0.empty() ) cout << "Press <space> to toggle background model update" << endl;
cout << "Press 's' to toggle foreground mask smoothing" << endl;
cout << "Press ESC or 'q' to exit" << endl;
bool doUpdateModel = true;
bool doSmoothMask = false;
Mat inputFrame, frame, foregroundMask, foreground, background;
for (;;)
{
// prepare input frame
cap >> inputFrame;
if (inputFrame.empty())
{
cout << "Finished reading: empty frame" << endl;
break; break;
resize(img0, img, Size(640, 640*img0.rows/img0.cols), 0, 0, INTER_LINEAR_EXACT);
if( fgimg.empty() )
fgimg.create(img.size(), img.type());
//update the model
bg_model->apply(img, fgmask, update_bg_model ? -1 : 0);
if( smoothMask )
{
GaussianBlur(fgmask, fgmask, Size(11, 11), 3.5, 3.5);
threshold(fgmask, fgmask, 10, 255, THRESH_BINARY);
} }
const Size scaledSize(640, 640 * inputFrame.rows / inputFrame.cols);
resize(inputFrame, frame, scaledSize, 0, 0, INTER_LINEAR);
fgimg = Scalar::all(0); // pass the frame to background model
img.copyTo(fgimg, fgmask); model->apply(frame, foregroundMask, doUpdateModel ? -1 : 0);
Mat bgimg; // show processed frame
bg_model->getBackgroundImage(bgimg); imshow("image", frame);
imshow("image", img); // show foreground image and mask (with optional smoothing)
imshow("foreground mask", fgmask); if (doSmoothMask)
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; GaussianBlur(foregroundMask, foregroundMask, Size(11, 11), 3.5, 3.5);
if(update_bg_model) threshold(foregroundMask, foregroundMask, 10, 255, THRESH_BINARY);
printf("Background update is on\n"); }
else if (foreground.empty())
printf("Background update is off\n"); foreground.create(scaledSize, frame.type());
foreground = Scalar::all(0);
frame.copyTo(foreground, foregroundMask);
imshow("foreground mask", foregroundMask);
imshow("foreground image", foreground);
// show background image
model->getBackgroundImage(background);
if (!background.empty())
imshow("mean background image", background );
// interact with user
const char key = (char)waitKey(30);
if (key == 27 || key == 'q') // ESC
{
cout << "Exit requested" << endl;
break;
}
else if (key == ' ')
{
doUpdateModel = !doUpdateModel;
cout << "Toggle background update: " << (doUpdateModel ? "ON" : "OFF") << endl;
}
else if (key == 's')
{
doSmoothMask = !doSmoothMask;
cout << "Toggle foreground mask smoothing: " << (doSmoothMask ? "ON" : "OFF") << endl;
} }
} }
return 0; return 0;
} }