FaceRecognizer: convert to grayscale only if 3 channels are provided

This commit is contained in:
Anshul Jain 2015-09-29 18:26:35 -07:00
parent a31402f60a
commit bb07fcdaf6

View File

@ -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;