bgslibrary v1.9.0

35 algorithms available

CMakeLists.txt

@@ -0,0 +1,52 @@
+
+cmake_minimum_required(VERSION 2.8)
+
+project(bgs)
+
+set(CMAKE_CXX_FLAGS "${CMAKE_CSS_FLAGS} -std=gnu++0x")
+
+find_package(OpenCV REQUIRED)
+
+if(${OpenCV_VERSION} VERSION_LESS 2.3.1)
+  message (FATAL_ERROR "OpenCV version is not compatible: ${OpenCV_VERSION}")
+endif()
+
+file(GLOB sources FrameProcessor.cpp PreProcessor.cpp VideoAnalysis.cpp  VideoCapture.cpp)
+file(GLOB main Main.cpp)
+file(GLOB demo Demo.cpp)
+
+list(REMOVE_ITEM sources ${demo})
+
+file(GLOB_RECURSE analysis package_analysis/*.cpp)
+file(GLOB_RECURSE bgs package_bgs/*.cpp)
+file(GLOB_RECURSE bgs_include package_bgs/*.h)
+
+# GMG is not available in older OpenCV versions
+if(${OpenCV_VERSION} VERSION_LESS 2.4.3)
+  file(GLOB gmg package_bgs/GMG.cpp)
+  list(REMOVE_ITEM bgs ${gmg})
+endif()
+
+include_directories(${CMAKE_SOURCE_DIR})
+
+add_library(bgs SHARED ${sources} ${bgs} ${analysis})
+target_link_libraries(bgs ${OpenCV_LIBS})
+set_property(TARGET bgs PROPERTY PUBLIC_HEADER ${bgs_include})
+
+add_executable(bgs_bin ${main})
+target_link_libraries(bgs_bin ${OpenCV_LIBS} bgs)
+set_target_properties(bgs_bin
+  PROPERTIES OUTPUT_NAME bgs)
+
+add_executable(bgs_demo ${demo})
+target_link_libraries(bgs_demo ${OpenCV_LIBS} bgs)
+
+INSTALL(TARGETS bgs
+	bgs_demo
+	bgs_bin
+  RUNTIME DESTINATION bin COMPONENT app
+  LIBRARY DESTINATION lib COMPONENT runtime
+  ARCHIVE DESTINATION lib COMPONENT runtime
+  PUBLIC_HEADER DESTINATION include/package_bgs COMPONENT dev
+  FRAMEWORK DESTINATION "/Library/Frameworks"
+)

Config.h

@@ -0,0 +1,22 @@
+/*
+This file is part of BGSLibrary.
+
+BGSLibrary is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+BGSLibrary is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with BGSLibrary.  If not, see <http://www.gnu.org/licenses/>.
+*/
+#pragma once
+
+const int KEY_REPEAT = 'r';
+const int KEY_SPACE  = 32;
+const int KEY_ESC    = 27;
+const int KEY_ESC2   = 'q';

Demo.cpp

@@ -0,0 +1,182 @@
+/*
+This file is part of BGSLibrary.
+
+BGSLibrary is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+BGSLibrary is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with BGSLibrary.  If not, see <http://www.gnu.org/licenses/>.
+*/
+#include <iostream>
+#include <cv.h>
+#include <highgui.h>
+
+#include "package_bgs/FrameDifferenceBGS.h"
+#include "package_bgs/StaticFrameDifferenceBGS.h"
+#include "package_bgs/WeightedMovingMeanBGS.h"
+#include "package_bgs/WeightedMovingVarianceBGS.h"
+#include "package_bgs/MixtureOfGaussianV1BGS.h"
+#include "package_bgs/MixtureOfGaussianV2BGS.h"
+#include "package_bgs/AdaptiveBackgroundLearning.h"
+#include "package_bgs/AdaptiveSelectiveBackgroundLearning.h"
+
+#if CV_MAJOR_VERSION >= 2 && CV_MINOR_VERSION >= 4 && CV_SUBMINOR_VERSION >= 3
+#include "package_bgs/GMG.h"
+#endif
+
+#include "package_bgs/dp/DPAdaptiveMedianBGS.h"
+#include "package_bgs/dp/DPGrimsonGMMBGS.h"
+#include "package_bgs/dp/DPZivkovicAGMMBGS.h"
+#include "package_bgs/dp/DPMeanBGS.h"
+#include "package_bgs/dp/DPWrenGABGS.h"
+#include "package_bgs/dp/DPPratiMediodBGS.h"
+#include "package_bgs/dp/DPEigenbackgroundBGS.h"
+#include "package_bgs/dp/DPTextureBGS.h"
+
+#include "package_bgs/tb/T2FGMM_UM.h"
+#include "package_bgs/tb/T2FGMM_UV.h"
+#include "package_bgs/tb/T2FMRF_UM.h"
+#include "package_bgs/tb/T2FMRF_UV.h"
+#include "package_bgs/tb/FuzzySugenoIntegral.h"
+#include "package_bgs/tb/FuzzyChoquetIntegral.h"
+
+#include "package_bgs/lb/LBSimpleGaussian.h"
+#include "package_bgs/lb/LBFuzzyGaussian.h"
+#include "package_bgs/lb/LBMixtureOfGaussians.h"
+#include "package_bgs/lb/LBAdaptiveSOM.h"
+#include "package_bgs/lb/LBFuzzyAdaptiveSOM.h"
+
+#if !defined(_WIN32)
+// Currently this method works only on Linux platform.
+#include "package_bgs/ck/LbpMrf.h"
+#endif
+
+#include "package_bgs/jmo/MultiLayerBGS.h"
+#include "package_bgs/pt/PixelBasedAdaptiveSegmenter.h"
+#include "package_bgs/av/VuMeter.h"
+#include "package_bgs/ae/KDE.h"
+#include "package_bgs/db/IndependentMultimodalBGS.h"
+#include "package_bgs/sjn/SJN_MultiCueBGS.h"
+
+void main(int argc, char **argv)
+{
+  std::cout << "Using OpenCV " << CV_MAJOR_VERSION << "." << CV_MINOR_VERSION << "." << CV_SUBMINOR_VERSION << std::endl;
+
+  CvCapture *capture = 0;
+  int resize_factor = 100;
+
+  if(argc > 1)
+  {
+    std::cout << "Openning: " << argv[1] << std::endl;
+    capture = cvCaptureFromAVI(argv[1]);
+  }
+  else
+  {
+    capture = cvCaptureFromCAM(0);
+    resize_factor = 50; // set size = 50% of original image
+  }
+
+  if(!capture)
+  {
+    std::cerr << "Cannot initialize video!" << std::endl;
+    return;
+  }
+  
+  IplImage *frame_aux = cvQueryFrame(capture);
+  IplImage *frame = cvCreateImage(cvSize((int)((frame_aux->width*resize_factor)/100) , (int)((frame_aux->height*resize_factor)/100)), frame_aux->depth, frame_aux->nChannels);
+  cvResize(frame_aux, frame);
+
+  /* Background Subtraction Methods */
+  IBGS *bgs;
+
+  /*** Default Package ***/
+  bgs = new FrameDifferenceBGS;
+  //bgs = new StaticFrameDifferenceBGS;
+  //bgs = new WeightedMovingMeanBGS;
+  //bgs = new WeightedMovingVarianceBGS;
+  //bgs = new MixtureOfGaussianV1BGS;
+  //bgs = new MixtureOfGaussianV2BGS;
+  //bgs = new AdaptiveBackgroundLearning;
+  //bgs = new AdaptiveSelectiveBackgroundLearning;
+  //bgs = new GMG;
+  
+  /*** DP Package (thanks to Donovan Parks) ***/
+  //bgs = new DPAdaptiveMedianBGS;
+  //bgs = new DPGrimsonGMMBGS;
+  //bgs = new DPZivkovicAGMMBGS;
+  //bgs = new DPMeanBGS;
+  //bgs = new DPWrenGABGS;
+  //bgs = new DPPratiMediodBGS;
+  //bgs = new DPEigenbackgroundBGS;
+  //bgs = new DPTextureBGS;
+
+  /*** TB Package (thanks to Thierry Bouwmans, Fida EL BAF and Zhenjie Zhao) ***/
+  //bgs = new T2FGMM_UM;
+  //bgs = new T2FGMM_UV;
+  //bgs = new T2FMRF_UM;
+  //bgs = new T2FMRF_UV;
+  //bgs = new FuzzySugenoIntegral;
+  //bgs = new FuzzyChoquetIntegral;
+
+  /*** JMO Package (thanks to Jean-Marc Odobez) ***/
+  //bgs = new MultiLayerBGS;
+
+  /*** PT Package (thanks to Martin Hofmann, Philipp Tiefenbacher and Gerhard Rigoll) ***/
+  //bgs = new PixelBasedAdaptiveSegmenter;
+
+  /*** LB Package (thanks to Laurence Bender) ***/
+  //bgs = new LBSimpleGaussian;
+  //bgs = new LBFuzzyGaussian;
+  //bgs = new LBMixtureOfGaussians;
+  //bgs = new LBAdaptiveSOM;
+  //bgs = new LBFuzzyAdaptiveSOM;
+
+  /*** LBP-MRF Package (thanks to Csaba Kertész) ***/
+  //bgs = new LbpMrf;
+
+  /*** AV Package (thanks to Lionel Robinault and Antoine Vacavant) ***/
+  //bgs = new VuMeter;
+
+  /*** EG Package (thanks to Ahmed Elgammal) ***/
+  //bgs = new KDE;
+  
+  /*** DB Package (thanks to Domenico Daniele Bloisi) ***/
+  //bgs = new IndependentMultimodalBGS;
+
+  /*** SJN Package (thanks to SeungJong Noh) ***/
+  //bgs = new SJN_MultiCueBGS;
+
+  int key = 0;
+  while(key != 'q')
+  {
+    frame_aux = cvQueryFrame(capture);
+    if(!frame_aux) break;
+
+    cvResize(frame_aux, frame);
+    
+    cv::Mat img_input(frame);
+    cv::imshow("input", img_input);
+
+    cv::Mat img_mask;
+    cv::Mat img_bkgmodel;
+    bgs->process(img_input, img_mask, img_bkgmodel); // by default, it shows automatically the foreground mask image
+    
+    //if(!img_mask.empty())
+    //  cv::imshow("Foreground", img_mask);
+    //  do something
+    
+    key = cvWaitKey(33);
+  }
+
+  delete bgs;
+
+  cvDestroyAllWindows();
+  cvReleaseCapture(&capture);
+}