mirror of
https://github.com/opencv/opencv.git
synced 2024-12-16 02:19:12 +08:00
124 lines
3.0 KiB
C++
124 lines
3.0 KiB
C++
#include <iostream>
|
|
#include <string>
|
|
|
|
#include "opencv2/core.hpp"
|
|
#include "opencv2/core/ocl.hpp"
|
|
#include "opencv2/core/utility.hpp"
|
|
#include "opencv2/videoio.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
#include "opencv2/video.hpp"
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
|
|
#define M_MOG2 2
|
|
#define M_KNN 3
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
CommandLineParser cmd(argc, argv,
|
|
"{ c camera | | use camera }"
|
|
"{ f file | ../data/vtest.avi | input video file }"
|
|
"{ t type | mog2 | method's type (knn, mog2) }"
|
|
"{ h help | | print help message }"
|
|
"{ m cpu_mode | false | press 'm' to switch OpenCL<->CPU}");
|
|
|
|
if (cmd.has("help"))
|
|
{
|
|
cout << "Usage : bgfg_segm [options]" << endl;
|
|
cout << "Available options:" << endl;
|
|
cmd.printMessage();
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
bool useCamera = cmd.has("camera");
|
|
string file = cmd.get<string>("file");
|
|
string method = cmd.get<string>("type");
|
|
|
|
if (method != "mog" && method != "mog2")
|
|
{
|
|
cerr << "Incorrect method" << endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
int m = method == "mog2" ? M_MOG2 : M_KNN;
|
|
|
|
VideoCapture cap;
|
|
if (useCamera)
|
|
cap.open(0);
|
|
else
|
|
cap.open(file);
|
|
|
|
if (!cap.isOpened())
|
|
{
|
|
cout << "can not open camera or video file" << endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
UMat frame, fgmask, fgimg;
|
|
cap >> frame;
|
|
fgimg.create(frame.size(), frame.type());
|
|
|
|
Ptr<BackgroundSubtractorKNN> knn = createBackgroundSubtractorKNN();
|
|
Ptr<BackgroundSubtractorMOG2> mog2 = createBackgroundSubtractorMOG2();
|
|
|
|
switch (m)
|
|
{
|
|
case M_KNN:
|
|
knn->apply(frame, fgmask);
|
|
break;
|
|
|
|
case M_MOG2:
|
|
mog2->apply(frame, fgmask);
|
|
break;
|
|
}
|
|
bool running=true;
|
|
for (;;)
|
|
{
|
|
if(!running)
|
|
break;
|
|
cap >> frame;
|
|
if (frame.empty())
|
|
break;
|
|
|
|
int64 start = getTickCount();
|
|
|
|
//update the model
|
|
switch (m)
|
|
{
|
|
case M_KNN:
|
|
knn->apply(frame, fgmask);
|
|
break;
|
|
|
|
case M_MOG2:
|
|
mog2->apply(frame, fgmask);
|
|
break;
|
|
}
|
|
|
|
double fps = getTickFrequency() / (getTickCount() - start);
|
|
std::cout << "FPS : " << fps << std::endl;
|
|
std::cout << fgimg.size() << std::endl;
|
|
fgimg.setTo(Scalar::all(0));
|
|
frame.copyTo(fgimg, fgmask);
|
|
|
|
imshow("image", frame);
|
|
imshow("foreground mask", fgmask);
|
|
imshow("foreground image", fgimg);
|
|
|
|
char key = (char)waitKey(30);
|
|
|
|
switch (key)
|
|
{
|
|
case 27:
|
|
running = false;
|
|
break;
|
|
case 'm':
|
|
case 'M':
|
|
ocl::setUseOpenCL(!ocl::useOpenCL());
|
|
cout << "Switched to " << (ocl::useOpenCL() ? "OpenCL enabled" : "CPU") << " mode\n";
|
|
break;
|
|
}
|
|
}
|
|
return EXIT_SUCCESS;
|
|
}
|