1 // minimalistic foreground-background segmentation sample, based off OpenCV's bgfg_segm sample
 2 
 3 #include "BackgroundSubtractorSuBSENSE.h"
 4 
 5 #include <opencv2/core/core.hpp>
 6 #include <opencv2/imgproc/imgproc.hpp>
 7 #include <opencv2/video/background_segm.hpp>
 8 #include <opencv2/highgui/highgui.hpp>
 9 #include <stdio.h>
10 
11 static void help() {
12     printf("\nMinimalistic example of foreground-background segmentation in a video sequence using\n"
13             "OpenCV's BackgroundSubtractor interface; will analyze frames from the default camera\n"
14             "or from a specified file.\n\n"
15             "Usage: \n"
16             "  ./bgfg_segm [--camera]=<use camera, true/false>, [--file]=<path to file> \n\n");
17 }
18 
19 const char* keys = {
20     "{c  |camera   |true     | use camera or not}"
21     "{f  |file     |tree.avi | movie file path  }"
22 };
23 
24 int main(int argc, const char** argv) {
25     help();
26     cv::CommandLineParser parser(argc, argv, keys);
27     const bool bUseDefaultCamera = parser.get<bool>("camera");
28     const std::string sVideoFilePath = parser.get<std::string>("file");
29     cv::VideoCapture oVideoInput;
30     cv::Mat oCurrInputFrame, oCurrSegmMask, oCurrReconstrBGImg;
31     if(bUseDefaultCamera) {
32         oVideoInput.open(0);
33         oVideoInput >> oCurrInputFrame;
34     }
35     else {
36         oVideoInput.open(sVideoFilePath);
37         oVideoInput >> oCurrInputFrame;
38         oVideoInput.set(CV_CAP_PROP_POS_FRAMES,0);
39     }
40     parser.printParams();
41     if(!oVideoInput.isOpened() || oCurrInputFrame.empty()) {
42         if(bUseDefaultCamera)
43             printf("Could not open default camera.\n");
44         else
45             printf("Could not open video file at '%s'.\n",sVideoFilePath.c_str());
46         return -1;
47     }
48     oCurrSegmMask.create(oCurrInputFrame.size(),CV_8UC1);
49     oCurrReconstrBGImg.create(oCurrInputFrame.size(),oCurrInputFrame.type());
50     cv::Mat oSequenceROI(oCurrInputFrame.size(),CV_8UC1,cv::Scalar_<uchar>(255)); // for optimal results, pass a constrained ROI to the algorithm (ex: for CDnet, use ROI.bmp)
51     cv::namedWindow("input",cv::WINDOW_NORMAL);
52     cv::namedWindow("segmentation mask",cv::WINDOW_NORMAL);
53     cv::namedWindow("reconstructed background",cv::WINDOW_NORMAL);
54     BackgroundSubtractorSuBSENSE oBGSAlg;
55     oBGSAlg.initialize(oCurrInputFrame,oSequenceROI);
56     for(int k=0;;++k) {
57         oVideoInput >> oCurrInputFrame;
58         if(oCurrInputFrame.empty())
59             break;
60         oBGSAlg(oCurrInputFrame,oCurrSegmMask,double(k<=100)); // lower rate in the early frames helps bootstrap the model when foreground is present
61         oBGSAlg.getBackgroundImage(oCurrReconstrBGImg);
62         imshow("input",oCurrInputFrame);
63         imshow("segmentation mask",oCurrSegmMask);
64         imshow("reconstructed background",oCurrReconstrBGImg);
65         if(cv::waitKey(1)==27)
66             break;
67     }
68     return 0;
69 }