From 3736e83aa0deeb8335f07d20c2939923dd125dd7 Mon Sep 17 00:00:00 2001
From: Anshul Jain <anshul@silklabs.com>
Date: Thu, 5 Nov 2015 12:08:43 +0530
Subject: [PATCH] Convert face training image to gray scale only if image has 3
 channels

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

diff --git a/src/FaceRecognizer.cc b/src/FaceRecognizer.cc
index 238a6af..9a2dd69 100644
--- a/src/FaceRecognizer.cc
+++ b/src/FaceRecognizer.cc
@@ -162,7 +162,9 @@ Local<Value> UnwrapTrainingData(Nan::NAN_METHOD_ARGS_TYPE info,
     int label = valarr->Get(0)->Uint32Value();
     cv::Mat im = fromMatrixOrFilename(valarr->Get(1));
     im = im.clone();
-    cv::cvtColor(im, im, CV_RGB2GRAY);
+    if (im.channels() == 3) {
+      cv::cvtColor(im, im, CV_RGB2GRAY);
+    }
     labels->push_back(label);
     images->push_back(im);
   }