refactored MOG2 algorithm
This commit is contained in:
@@ -76,8 +76,8 @@ int main(int argc, const char** argv)
|
||||
GpuMat d_frame(frame);
|
||||
|
||||
FGDStatModel fgd_stat;
|
||||
cv::Ptr<cv::BackgroundSubtractorMOG> mog = cv::gpu::createBackgroundSubtractorMOG();
|
||||
MOG2_GPU mog2;
|
||||
cv::Ptr<cv::BackgroundSubtractor> mog = cv::gpu::createBackgroundSubtractorMOG();
|
||||
cv::Ptr<cv::BackgroundSubtractor> mog2 = cv::gpu::createBackgroundSubtractorMOG2();
|
||||
GMG_GPU gmg;
|
||||
gmg.numInitializationFrames = 40;
|
||||
|
||||
@@ -100,7 +100,7 @@ int main(int argc, const char** argv)
|
||||
break;
|
||||
|
||||
case MOG2:
|
||||
mog2(d_frame, d_fgmask);
|
||||
mog2->apply(d_frame, d_fgmask);
|
||||
break;
|
||||
|
||||
case GMG:
|
||||
@@ -140,8 +140,8 @@ int main(int argc, const char** argv)
|
||||
break;
|
||||
|
||||
case MOG2:
|
||||
mog2(d_frame, d_fgmask);
|
||||
mog2.getBackgroundImage(d_bgimg);
|
||||
mog2->apply(d_frame, d_fgmask);
|
||||
mog2->getBackgroundImage(d_bgimg);
|
||||
break;
|
||||
|
||||
case GMG:
|
||||
|
@@ -1399,13 +1399,13 @@ TEST(MOG2)
|
||||
|
||||
cap >> frame;
|
||||
|
||||
cv::Ptr<cv::BackgroundSubtractor> d_mog2 = cv::gpu::createBackgroundSubtractorMOG2();
|
||||
cv::gpu::GpuMat d_frame(frame);
|
||||
cv::gpu::MOG2_GPU d_mog2;
|
||||
cv::gpu::GpuMat d_foreground;
|
||||
cv::gpu::GpuMat d_background;
|
||||
|
||||
d_mog2(d_frame, d_foreground);
|
||||
d_mog2.getBackgroundImage(d_background);
|
||||
d_mog2->apply(d_frame, d_foreground);
|
||||
d_mog2->getBackgroundImage(d_background);
|
||||
|
||||
while (!TestSystem::instance().stop())
|
||||
{
|
||||
@@ -1414,8 +1414,8 @@ TEST(MOG2)
|
||||
|
||||
TestSystem::instance().gpuOn();
|
||||
|
||||
d_mog2(d_frame, d_foreground);
|
||||
d_mog2.getBackgroundImage(d_background);
|
||||
d_mog2->apply(d_frame, d_foreground);
|
||||
d_mog2->getBackgroundImage(d_background);
|
||||
|
||||
TestSystem::instance().gpuOff();
|
||||
}
|
||||
|
Reference in New Issue
Block a user