FeaturesMain.cpp

/*
 Copyright 2012 Andrew Perrault and Saurav Kumar.

 This file is part of DetectText.

 DetectText is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 DetectText is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with DetectText. If not, see <http://www.gnu.org/licenses/>.
*/
#include <cassert>
#include <fstream>
#include <exception>
#include <iostream>

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>

#include "TextDetection.h"

using namespace std;
using namespace cv;
using namespace DetectText;

void convertToFloatImage ( Mat& byteImage, Mat& floatImage )
{
 byteImage.convertTo(floatImage, CV_32FC1, 1 / 255.);
}

class FeatureError: public std::exception {
 std::string message;
public:
 FeatureError(const std::string & msg, const std::string & file) {
 std::stringstream ss;

 ss << msg << " " << file;
 message = msg.c_str();
 }
 ~FeatureError() throw () {
 }
};

Mat loadByteImage(const char * name) {
 Mat image = imread(name);

 if (image.empty()) {
 return Mat();
 }
 cvtColor(image, image, CV_BGR2RGB);
 return image;
}

Mat loadFloatImage(const char * name) {
 Mat image = imread(name);

 if (image.empty()) {
 return Mat();
 }
 cvtColor(image, image, CV_BGR2RGB);
 Mat floatingImage(image.size(), CV_32FC3);
 image.convertTo(floatingImage, CV_32F, 1 / 255.);
 return floatingImage;
}

int mainTextDetection(int argc, char** argv) {
 Mat byteQueryImage = loadByteImage(argv[1]);
 if (byteQueryImage.empty()) {
 cerr << "couldn't load query image" << endl;
 return -1;
 }

 // Detect text in the image
 Mat output = textDetection(byteQueryImage, atoi(argv[3]));
 imwrite(argv[2], output);
 return 0;
}

int main(int argc, char** argv) {
 if ((argc != 4)) {
 cerr << "usage: " << argv[0] << " imagefile resultImage darkText" << endl;
 return -1;
 }
 return mainTextDetection(argc, argv);
}