mirror of
https://github.com/opencv/opencv.git
synced 2024-11-30 06:10:02 +08:00
174 lines
3.8 KiB
C++
174 lines
3.8 KiB
C++
#include <iostream>
|
|
#include <string>
|
|
|
|
#include "opencv2/core.hpp"
|
|
#include "opencv2/core/utility.hpp"
|
|
#include "opencv2/gpu.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
|
|
#include "opencv2/opencv_modules.hpp"
|
|
|
|
#ifdef HAVE_OPENCV_NONFREE
|
|
# include "opencv2/nonfree/gpu.hpp"
|
|
#endif
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
using namespace cv::gpu;
|
|
|
|
enum Method
|
|
{
|
|
MOG,
|
|
MOG2,
|
|
GMG,
|
|
FGD_STAT
|
|
};
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
cv::CommandLineParser cmd(argc, argv,
|
|
"{ c camera | | use camera }"
|
|
"{ f file | 768x576.avi | input video file }"
|
|
"{ m method | mog | method (mog, mog2, gmg, fgd) }"
|
|
"{ h help | | print help message }");
|
|
|
|
if (cmd.has("help") || !cmd.check())
|
|
{
|
|
cmd.printMessage();
|
|
cmd.printErrors();
|
|
return 0;
|
|
}
|
|
|
|
bool useCamera = cmd.has("camera");
|
|
string file = cmd.get<string>("file");
|
|
string method = cmd.get<string>("method");
|
|
|
|
if (method != "mog"
|
|
&& method != "mog2"
|
|
&& method != "gmg"
|
|
&& method != "fgd")
|
|
{
|
|
cerr << "Incorrect method" << endl;
|
|
return -1;
|
|
}
|
|
|
|
Method m = method == "mog" ? MOG :
|
|
method == "mog2" ? MOG2 :
|
|
method == "fgd" ? FGD_STAT :
|
|
GMG;
|
|
|
|
VideoCapture cap;
|
|
|
|
if (useCamera)
|
|
cap.open(0);
|
|
else
|
|
cap.open(file);
|
|
|
|
if (!cap.isOpened())
|
|
{
|
|
cerr << "can not open camera or video file" << endl;
|
|
return -1;
|
|
}
|
|
|
|
Mat frame;
|
|
cap >> frame;
|
|
|
|
GpuMat d_frame(frame);
|
|
|
|
Ptr<BackgroundSubtractor> mog = gpu::createBackgroundSubtractorMOG();
|
|
Ptr<BackgroundSubtractor> mog2 = gpu::createBackgroundSubtractorMOG2();
|
|
Ptr<BackgroundSubtractor> gmg = gpu::createBackgroundSubtractorGMG(40);
|
|
Ptr<BackgroundSubtractor> fgd = gpu::createBackgroundSubtractorFGD();
|
|
|
|
GpuMat d_fgmask;
|
|
GpuMat d_fgimg;
|
|
GpuMat d_bgimg;
|
|
|
|
Mat fgmask;
|
|
Mat fgimg;
|
|
Mat bgimg;
|
|
|
|
switch (m)
|
|
{
|
|
case MOG:
|
|
mog->apply(d_frame, d_fgmask, 0.01);
|
|
break;
|
|
|
|
case MOG2:
|
|
mog2->apply(d_frame, d_fgmask);
|
|
break;
|
|
|
|
case GMG:
|
|
gmg->apply(d_frame, d_fgmask);
|
|
break;
|
|
|
|
case FGD_STAT:
|
|
fgd->apply(d_frame, d_fgmask);
|
|
break;
|
|
}
|
|
|
|
namedWindow("image", WINDOW_NORMAL);
|
|
namedWindow("foreground mask", WINDOW_NORMAL);
|
|
namedWindow("foreground image", WINDOW_NORMAL);
|
|
if (m != GMG)
|
|
{
|
|
namedWindow("mean background image", WINDOW_NORMAL);
|
|
}
|
|
|
|
for(;;)
|
|
{
|
|
cap >> frame;
|
|
if (frame.empty())
|
|
break;
|
|
d_frame.upload(frame);
|
|
|
|
int64 start = cv::getTickCount();
|
|
|
|
//update the model
|
|
switch (m)
|
|
{
|
|
case MOG:
|
|
mog->apply(d_frame, d_fgmask, 0.01);
|
|
mog->getBackgroundImage(d_bgimg);
|
|
break;
|
|
|
|
case MOG2:
|
|
mog2->apply(d_frame, d_fgmask);
|
|
mog2->getBackgroundImage(d_bgimg);
|
|
break;
|
|
|
|
case GMG:
|
|
gmg->apply(d_frame, d_fgmask);
|
|
break;
|
|
|
|
case FGD_STAT:
|
|
fgd->apply(d_frame, d_fgmask);
|
|
fgd->getBackgroundImage(d_bgimg);
|
|
break;
|
|
}
|
|
|
|
double fps = cv::getTickFrequency() / (cv::getTickCount() - start);
|
|
std::cout << "FPS : " << fps << std::endl;
|
|
|
|
d_fgimg.setTo(Scalar::all(0));
|
|
d_frame.copyTo(d_fgimg, d_fgmask);
|
|
|
|
d_fgmask.download(fgmask);
|
|
d_fgimg.download(fgimg);
|
|
if (!d_bgimg.empty())
|
|
d_bgimg.download(bgimg);
|
|
|
|
imshow("image", frame);
|
|
imshow("foreground mask", fgmask);
|
|
imshow("foreground image", fgimg);
|
|
if (!bgimg.empty())
|
|
imshow("mean background image", bgimg);
|
|
|
|
int key = waitKey(30);
|
|
if (key == 27)
|
|
break;
|
|
}
|
|
|
|
return 0;
|
|
}
|