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);
}