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