From 0dbf8e7f05942e4bf1908e99d0e4eef060207314 Mon Sep 17 00:00:00 2001
From: Micooz <Micooz@hotmail.com>
Date: Wed, 12 Aug 2015 15:21:54 +0800
Subject: [PATCH] Fixed bug: char blocks must be read as grayscale images.

---
 src/train/ann_train.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/train/ann_train.cpp b/src/train/ann_train.cpp
index 02a567c..6ceb138 100644
--- a/src/train/ann_train.cpp
+++ b/src/train/ann_train.cpp
@@ -53,7 +53,7 @@ cv::Ptr<cv::ml::TrainData> AnnTrain::train_data() {
 
     auto chars_files = utils::getFiles(sub_folder);
     for (auto file : chars_files) {
-      auto img = cv::imread(file);
+      auto img = cv::imread(file, 0); // a grayscale image
       auto fps = features(img, kPredictSize);
 
       samples.push_back(fps);
@@ -91,7 +91,7 @@ void AnnTrain::test() {
     std::vector<std::pair<std::string, std::string>> error_files;
 
     for (auto file : chars_files) {
-      auto img = cv::imread(file);
+      auto img = cv::imread(file, 0); // a grayscale image
       std::pair<std::string, std::string> ch = CharsIdentify::instance()->identify(img);
       if (ch.first == char_key) {
         ++corrects;