bgsegm: convert rgb images to gray in CNT

8 years ago · d253a0fa13
parent 34f910bb72
commit d253a0fa13
2 changed files with 4 additions and 1 deletions
--- a/modules/bgsegm/samples/bgfg_gmg.cpp
+++ b/modules/bgsegm/samples/bgfg_gmg.cpp
--- a/modules/bgsegm/src/bgfg_subcnt.cpp
+++ b/modules/bgsegm/src/bgfg_subcnt.cpp
@ -341,9 +341,12 @@ private:

 void BackgroundSubtractorCNTImpl::apply(InputArray image, OutputArray _fgmask, double learningRate)
 {
-    CV_Assert(image.type() == CV_8UC1);
+    CV_Assert(image.depth() == CV_8U);

    Mat frameIn = image.getMat();
+    if(frameIn.channels() != 1)
+        cvtColor(frameIn, frameIn, COLOR_BGR2GRAY);
+
    _fgmask.create(image.size(), CV_8U); // OutputArray usage requires this step
    Mat fgMask = _fgmask.getMat();