bgsegm: convert rgb images to gray in CNT

d253a0fa · Vladislav Sovrasov · 34f910bb · d253a0fa · d253a0fa
Commit d253a0fa authored Mar 14, 2017 by Vladislav Sovrasov
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

bgfg.cpp modules/bgsegm/samples/bgfg.cpp +0 -0

bgfg_subcnt.cpp modules/bgsegm/src/bgfg_subcnt.cpp +4 -1

No files found.
--- 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();