generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added GMG_GPU to bgfg_segm sample

Browse Source
This commit is contained in:
Vladislav Vinogradov 2012-08-09 12:59:18 +04:00
parent f7f1fb2bd7
commit beb3e6ff18
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
samples/gpu/bgfg_segm.cpp
View File

@ -14,7 +14,8 @@ enum Method
FGD_STAT, FGD_STAT,
MOG, MOG,
MOG2, MOG2,
VIBE VIBE,
GMG
}; };
int main(int argc, const char** argv) int main(int argc, const char** argv)
@ -22,7 +23,7 @@ int main(int argc, const char** argv)
cv::CommandLineParser cmd(argc, argv, cv::CommandLineParser cmd(argc, argv,
"{ c | camera | false | use camera }" "{ c | camera | false | use camera }"
"{ f | file | 768x576.avi | input video file }" "{ f | file | 768x576.avi | input video file }"
"{ m | method | mog | method (fgd_stat, mog, mog2, vibe) }" "{ m | method | mog | method (fgd, mog, mog2, vibe, gmg) }"
"{ h | help | false | print help message }"); "{ h | help | false | print help message }");
if (cmd.get<bool>("help")) if (cmd.get<bool>("help"))
@ -37,13 +38,13 @@ int main(int argc, const char** argv)
string file = cmd.get<string>("file"); string file = cmd.get<string>("file");
string method = cmd.get<string>("method"); string method = cmd.get<string>("method");
if (method != "fgd_stat" && method != "mog" && method != "mog2" && method != "vibe") if (method != "fgd" && method != "mog" && method != "mog2" && method != "vibe" && method != "gmg")
{ {
cerr << "Incorrect method" << endl; cerr << "Incorrect method" << endl;
return -1; return -1;
} }
Method m = method == "fgd_stat" ? FGD_STAT : method == "mog" ? MOG : method == "mog2" ? MOG2 : VIBE; Method m = method == "fgd" ? FGD_STAT : method == "mog" ? MOG : method == "mog2" ? MOG2 : method == "vibe" ? VIBE : GMG;
VideoCapture cap; VideoCapture cap;
@ -67,6 +68,8 @@ int main(int argc, const char** argv)
MOG_GPU mog; MOG_GPU mog;
MOG2_GPU mog2; MOG2_GPU mog2;
VIBE_GPU vibe; VIBE_GPU vibe;
GMG_GPU gmg;
gmg.numInitializationFrames = 40;
GpuMat d_fgmask; GpuMat d_fgmask;
GpuMat d_fgimg; GpuMat d_fgimg;
@ -93,12 +96,16 @@ int main(int argc, const char** argv)
case VIBE: case VIBE:
vibe.initialize(d_frame); vibe.initialize(d_frame);
break; break;
case GMG:
gmg.initialize(d_frame.size());
break;
} }
namedWindow("image", WINDOW_NORMAL); namedWindow("image", WINDOW_NORMAL);
namedWindow("foreground mask", WINDOW_NORMAL); namedWindow("foreground mask", WINDOW_NORMAL);
namedWindow("foreground image", WINDOW_NORMAL); namedWindow("foreground image", WINDOW_NORMAL);
if (m != VIBE) if (m != VIBE && m != GMG)
namedWindow("mean background image", WINDOW_NORMAL); namedWindow("mean background image", WINDOW_NORMAL);
for(;;) for(;;)
@ -108,6 +115,8 @@ int main(int argc, const char** argv)
break; break;
d_frame.upload(frame); d_frame.upload(frame);
int64 start = cv::getTickCount();
//update the model //update the model
switch (m) switch (m)
{ {
@ -130,8 +139,15 @@ int main(int argc, const char** argv)
case VIBE: case VIBE:
vibe(d_frame, d_fgmask); vibe(d_frame, d_fgmask);
break; 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_fgimg.setTo(Scalar::all(0));
d_frame.copyTo(d_fgimg, d_fgmask); d_frame.copyTo(d_fgimg, d_fgmask);