mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
171 lines
3.7 KiB
171 lines
3.7 KiB
#include <iostream> |
|
#include <string> |
|
|
|
#include "opencv2/core/core.hpp" |
|
#include "opencv2/gpu/gpu.hpp" |
|
#include "opencv2/highgui/highgui.hpp" |
|
|
|
using namespace std; |
|
using namespace cv; |
|
using namespace cv::gpu; |
|
|
|
enum Method |
|
{ |
|
FGD_STAT, |
|
MOG, |
|
MOG2, |
|
VIBE, |
|
GMG |
|
}; |
|
|
|
int main(int argc, const char** argv) |
|
{ |
|
cv::CommandLineParser cmd(argc, argv, |
|
"{ c | camera | false | use camera }" |
|
"{ f | file | 768x576.avi | input video file }" |
|
"{ m | method | mog | method (fgd, mog, mog2, vibe, gmg) }" |
|
"{ h | help | false | print help message }"); |
|
|
|
if (cmd.get<bool>("help")) |
|
{ |
|
cout << "Usage : bgfg_segm [options]" << endl; |
|
cout << "Avaible options:" << endl; |
|
cmd.printParams(); |
|
return 0; |
|
} |
|
|
|
bool useCamera = cmd.get<bool>("camera"); |
|
string file = cmd.get<string>("file"); |
|
string method = cmd.get<string>("method"); |
|
|
|
if (method != "fgd" && method != "mog" && method != "mog2" && method != "vibe" && method != "gmg") |
|
{ |
|
cerr << "Incorrect method" << endl; |
|
return -1; |
|
} |
|
|
|
Method m = method == "fgd" ? FGD_STAT : method == "mog" ? MOG : method == "mog2" ? MOG2 : method == "vibe" ? VIBE : 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); |
|
|
|
FGDStatModel fgd_stat; |
|
MOG_GPU mog; |
|
MOG2_GPU mog2; |
|
VIBE_GPU vibe; |
|
GMG_GPU gmg; |
|
gmg.numInitializationFrames = 40; |
|
|
|
GpuMat d_fgmask; |
|
GpuMat d_fgimg; |
|
GpuMat d_bgimg; |
|
|
|
Mat fgmask; |
|
Mat fgimg; |
|
Mat bgimg; |
|
|
|
switch (m) |
|
{ |
|
case FGD_STAT: |
|
fgd_stat.create(d_frame); |
|
break; |
|
|
|
case MOG: |
|
mog(d_frame, d_fgmask, 0.01f); |
|
break; |
|
|
|
case MOG2: |
|
mog2(d_frame, d_fgmask); |
|
break; |
|
|
|
case VIBE: |
|
vibe.initialize(d_frame); |
|
break; |
|
|
|
case GMG: |
|
gmg.initialize(d_frame.size()); |
|
break; |
|
} |
|
|
|
namedWindow("image", WINDOW_NORMAL); |
|
namedWindow("foreground mask", WINDOW_NORMAL); |
|
namedWindow("foreground image", WINDOW_NORMAL); |
|
if (m != VIBE && 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 FGD_STAT: |
|
fgd_stat.update(d_frame); |
|
d_fgmask = fgd_stat.foreground; |
|
d_bgimg = fgd_stat.background; |
|
break; |
|
|
|
case MOG: |
|
mog(d_frame, d_fgmask, 0.01f); |
|
mog.getBackgroundImage(d_bgimg); |
|
break; |
|
|
|
case MOG2: |
|
mog2(d_frame, d_fgmask); |
|
mog2.getBackgroundImage(d_bgimg); |
|
break; |
|
|
|
case VIBE: |
|
vibe(d_frame, d_fgmask); |
|
break; |
|
|
|
case GMG: |
|
gmg(d_frame, d_fgmask); |
|
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; |
|
}
|
|
|