1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
/*-----------------------------------------------------------------
LOG
GEM - Graphics Environment for Multimedia
Threshold filter
Copyright (c) 1997-1999 Mark Danks. mark@danks.org
Copyright (c) Günther Geiger. geiger@epy.co.at
Copyright (c) 2001-2002 IOhannes m zmoelnig. forum::für::umläute. IEM. zmoelnig@iem.kug.ac.at
Copyright (c) 2002 James Tittle & Chris Clepper
For information on usage and redistribution, and for a DISCLAIMER OF ALL
WARRANTIES, see the file, "GEM.LICENSE.TERMS" in this distribution.
-----------------------------------------------------------------*/
#ifndef INCLUDE_PIX_OPENCV_BACKGROUNDSUBTRACTOR_H_
#define INCLUDE_PIX_OPENCV_BACKGROUNDSUBTRACTOR_H_
#include "opencv2/opencv.hpp"
#if HAVE_LIBOPENCV_CL
#include "opencv2/ocl/ocl.hpp"
#endif /* HAVE_LIBOPENCV_CL */
#include "Base/GemPixObj.h"
#include <RTE/MessageCallbacks.h>
#include "Gem/Exception.h"
#include <iostream>
/*-----------------------------------------------------------------
-------------------------------------------------------------------
CLASS
pix_opencv_backgroundsubtractor
generic dynamic background subtractor, should implement severals algo
KEYWORDS
pix
DESCRIPTION
-----------------------------------------------------------------*/
class GEM_EXPORT pix_opencv_backgroundsubtractor: public GemPixObj
{
CPPEXTERN_HEADER(pix_opencv_backgroundsubtractor, GemPixObj)
public:
//////////
// Constructor
pix_opencv_backgroundsubtractor();
protected:
// Message handling
void enumParamsMess();
void setParamMess(t_symbol *s, int argc, t_atom* argv);
void getParamMess(t_symbol *paramName);
void paramHelpMess();
void algoMess(t_symbol *s, int argc, t_atom* argv);
void cpuModeMess(int val);
//////////
// Destructor
virtual ~pix_opencv_backgroundsubtractor();
//////////
// Do the processing
virtual void processImage(imageStruct &image);
virtual void startRendering();
virtual void stopRendering();
private:
cv::Ptr<cv::BackgroundSubtractor> m_fgbgMOG;
cv::Ptr<cv::BackgroundSubtractorGMG> m_fgbgGMG;
#if HAVE_LIBOPENCV_CL
cv::ocl::MOG m_oclMOG;
cv::ocl::MOG2 m_oclMOG2;
//~ cv::ocl::GMG m_oclGMG;
cv::ocl::oclMat d_input, d_fgmask;
#endif /* HAVE_LIBOPENCV_CL */
cv::Mat m_fgmask, m_segm;
std::string m_algoName;
t_outlet *m_dataout;
t_float m_threshold, m_initFrames;
bool m_rendering, m_gpuMode, m_forceCPU;
std::vector<std::string> m_bgsub_algos;
};
#endif // for header file
|
