controller.cpp modified

Author: dkostyrev

main.cpp

@@ -45,6 +45,7 @@ int main()
     Controller controller = Controller();
     while (cap.read(frame)) {
         controller.Process(frame);
+        cv::imshow("frame", frame);
         int key = cv::waitKey(1);
         if (key == 27)
             return 0;

src/controller.cpp

@@ -33,7 +33,7 @@ void Controller::Process(cv::Mat frame)
         return;
 
     if (threshold == 0) {
-        threshold = static_cast<int>(frame.rows * frame.cols * 0.1);
+        threshold = static_cast<int>(frame.rows * frame.cols * 0.05);
         std::cout << "Threshold = " << threshold << std::endl;
     }
     if (topthreshold == 0) {

src/vibe.cpp

@@ -12,8 +12,8 @@ ViBe::ViBe(int width, int height)
 
 void ViBe::Initialize(cv::Mat firstframe)
 {
-    N = 10;
-    R = 20;
+    N = 2;
+    R = 40;
     smin = 1;
     r_s = 20 ;
     samples = std::vector<cv::Mat>(N);
@@ -25,6 +25,7 @@ void ViBe::Initialize(cv::Mat firstframe)
 
 void ViBe::Process(cv::Mat image, cv::Mat& segMap)
 {
+    segMap = cv::Mat(image.size(), CV_8UC1);
     for (int x = 0; x < image.cols ; x++){
         for (int y = 0 ; y < image.rows; y++) {
             int count = 0, index = 0 , dist = 0;