From bb07fcdaf6d60d758eadb113e84f77e217ee0aed Mon Sep 17 00:00:00 2001
From: Anshul Jain <anshul@silklabs.com>
Date: Tue, 29 Sep 2015 18:26:35 -0700
Subject: [PATCH] FaceRecognizer: convert to grayscale only if 3 channels are
 provided

---
 src/FaceRecognizer.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/FaceRecognizer.cc b/src/FaceRecognizer.cc
index 8bb1bfa..8de89c8 100644
--- a/src/FaceRecognizer.cc
+++ b/src/FaceRecognizer.cc
@@ -213,7 +213,10 @@ NAN_METHOD(FaceRecognizerWrap::PredictSync) {
   SETUP_FUNCTION(FaceRecognizerWrap)
 
   cv::Mat im = fromMatrixOrFilename(info[0]);  // TODO CHECK!
-  cv::cvtColor(im, im, CV_RGB2GRAY);
+  if (im.channels() == 3) {
+    cv::cvtColor(im, im, CV_RGB2GRAY);
+  }
+
   // int predictedLabel = self->rec->predict(im);
 
   int predictedLabel = -1;