Fixes for CodeBook algorithm

Demo.py

@@ -45,6 +45,7 @@
 #bgs = libbgs.WeightedMovingVariance()
 #bgs = libbgs.TwoPoints()
 #bgs = libbgs.ViBe()
+#bgs = libbgs.CodeBook()
 
 video_file = "dataset/video.avi"
 

VideoCapture.cpp

@@ -135,14 +135,18 @@ namespace bgslibrary
     if (useVideo)  setUpVideo();
     //if (!capture)  std::cerr << "Capture error..." << std::endl;
 
-    using namespace std::chrono_literals;
+    //using namespace std::chrono_literals;
     do
     {
       capture >> frame;
       if (frame.empty())
       {
         std::cout << "Frame is not ready" << std::endl;
-        std::this_thread::sleep_for(1s);
+        std::string dummy;
+        std::cout << "Enter to continue..." << std::endl;
+        std::getline(std::cin, dummy);
+        //cv::waitKey(1000);
+        //std::this_thread::sleep_for(1s);
       }
       else
         break;
@@ -193,7 +197,7 @@ namespace bgslibrary
             std::cout << "Set ROI (press ESC to skip)" << std::endl;
             VC_ROI::img_input1 = new IplImage(img_input);
             cvSetMouseCallback("Input", VC_ROI::VideoCapture_on_mouse, NULL);
-            key = cvWaitKey(0);
+            key = cv::waitKey(0);
             delete VC_ROI::img_input1;
           }
           else
@@ -239,17 +243,17 @@ namespace bgslibrary
 
       //cvResetImageROI(frame);
 
-      key = cvWaitKey(loopDelay);
+      key = cv::waitKey(loopDelay);
       //std::cout << "key: " << key << std::endl;
 
       if (key == KEY_SPACE)
-        key = cvWaitKey(0);
+        key = cv::waitKey(0);
 
       if (key == KEY_ESC)
         break;
 
       if (stopAt > 0 && stopAt == frameNumber)
-        key = cvWaitKey(0);
+        key = cv::waitKey(0);
 
       firstTime = false;
     } while (1);

VideoCapture.h

@@ -17,8 +17,9 @@ along with BGSLibrary.  If not, see <http://www.gnu.org/licenses/>.
 #pragma once
 
 #include <iostream>
-#include <chrono>
-#include <thread>
+#include <fstream>
+//#include <chrono>
+//#include <thread>
 #include <opencv2/opencv.hpp>
 #include <opencv2/imgproc/imgproc_c.h>
 #include <opencv2/imgproc/types_c.h>

wrapper_matlab/backgroundSubtractor_wrapper.cpp

@@ -66,6 +66,7 @@
 #include "PAWCS.h"
 #include "TwoPoints.h"
 #include "ViBe.h"
+#include "CodeBook.h"
 
 using namespace bgslibrary::algorithms;
 
@@ -165,6 +166,8 @@ namespace bgslibrary
       return (IBGS *)mxCalloc(1, sizeof(TwoPoints));
     if (alg_name.compare("ViBe") == 0)
       return (IBGS *)mxCalloc(1, sizeof(ViBe));
+    if (alg_name.compare("CodeBook") == 0)
+      return (IBGS *)mxCalloc(1, sizeof(CodeBook));
     return NULL;
   }
 
@@ -260,6 +263,8 @@ namespace bgslibrary
       return new (ptrBGS) TwoPoints();
     if (alg_name.compare("ViBe") == 0)
       return new (ptrBGS) ViBe();
+    if (alg_name.compare("CodeBook") == 0)
+      return new (ptrBGS) CodeBook();
     return NULL;
   }
 }

wrapper_matlab/compile.m

@@ -91,4 +91,5 @@
   "../package_bgs/ViBe.cpp" ...
   "../package_bgs/ViBe/vibe-background-sequential.cpp" ...
   "../package_bgs/TwoPoints.cpp" ...
-  "../package_bgs/TwoPoints/two_points.cpp"
+  "../package_bgs/TwoPoints/two_points.cpp" ...
+  "../package_bgs/CodeBook.cpp"

wrapper_matlab/run_demo.m

@@ -56,4 +56,5 @@
 % demo('LOBSTER')
 % demo('PAWCS')
 % demo('TwoPoints')
-% demo('ViBe')
+% demo('ViBe')
+% demo('CodeBook')

wrapper_python/bgslibrary_module.cpp

@@ -260,6 +260,11 @@ namespace fs
 				.def("apply", &ViBe::apply)
 				.def("getBackgroundModel", &ViBe::getBackgroundModel)
 				;
+
+			py::class_<CodeBook>("CodeBook")
+				.def("apply", &CodeBook::apply)
+				.def("getBackgroundModel", &CodeBook::getBackgroundModel)
+				;
 		}
 	} // namespace fs
 } // namespace python