From 2ee96aed15baaa13cf7c1b2b127ebff5aacb2953 Mon Sep 17 00:00:00 2001
From: Guangli Li <guangli.li@hotmail.com>
Date: Thu, 7 Dec 2017 23:07:32 -0500
Subject: [PATCH] Fixes for CodeBook algorithm

---
 package_bgs/CodeBook.cpp | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/package_bgs/CodeBook.cpp b/package_bgs/CodeBook.cpp
index a1e4056..5b066ef 100644
--- a/package_bgs/CodeBook.cpp
+++ b/package_bgs/CodeBook.cpp
@@ -43,7 +43,14 @@ void CodeBook::process(const cv::Mat &img_input, cv::Mat &img_output, cv::Mat &i
   } 
 
   cv::Mat img_input_gray;
-  cv::cvtColor(img_input, img_input_gray, CV_BGR2GRAY);
+
+  if (img_input.channels() == 1)
+  {
+    img_input_gray = img_input; 
+  } else
+  { 
+    cv::cvtColor(img_input, img_input_gray, CV_BGR2GRAY);
+  }
 
   fg_cb(img_input_gray, img_foreground);
 
-- 
GitLab