//Default constructor
OCRTesseractImpl(const char* datapath, const char* language, const char* char_whitelist, int oemode, int psmode)
{
#ifdef HAVE_TESSERACT
const char *lang = "eng";
if (language != NULL)
lang = language;
if (tess.Init(datapath, lang, (tesseract::OcrEngineMode)oemode))
{
cout << "OCRTesseract: Could not initialize tesseract." << endl;
throw 1;
}
//cout << "OCRTesseract: tesseract version " << tess.Version() << endl;
tesseract::PageSegMode pagesegmode = (tesseract::PageSegMode)psmode;
tess.SetPageSegMode(pagesegmode);
if(char_whitelist != NULL)
tess.SetVariable("tessedit_char_whitelist", char_whitelist);
else
tess.SetVariable("tessedit_char_whitelist", "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ");
tess.SetVariable("save_best_choices", "T");
#else
cout << "OCRTesseract("<<oemode<<psmode<<"): Tesseract not found." << endl;
if (datapath != NULL)
cout << " " << datapath << endl;
if (language != NULL)
cout << " " << language << endl;
if (char_whitelist != NULL)
cout << " " << char_whitelist << endl;
#endif
}
int main(int argc, char *argv[])
{
if(argc < 2)
{
cout << "Usage: main <image-file-name>" << endl;
exit(0);
}
// load the image
Mat img = imread(argv[1]);
//references = getReferences();//load the reference images
tess.Init(NULL, NULL, tesseract::OEM_DEFAULT);
tess.SetVariable("tessedit_char_whitelist", "0123456789");
//for speed testing
/*double t0 = (double) getTickCount();/
for(int i = 0; i < 100; i++)
{
signDetection(img);
}
double t1 = (double) getTickCount();
cout << (t1-t0) / getTickFrequency() << endl;*/
signDetection(img);
waitKey(0);
return 0;
}
/*
** ocr_type=0: OEM_DEFAULT
** ocr_type=1: OEM_TESSERACT_ONLY
** ocr_type=2: OEM_CUBE_ONLY
** ocr_type=3: OEM_TESSERACT_CUBE_COMBINED
*/
int tess_capi_init(char *datapath,char *language,int ocr_type,FILE *out,
char *initstr,int maxlen)
{
int status;
#ifdef USE_NLS
setlocale (LC_ALL, "");
bindtextdomain (PACKAGE, LOCALEDIR);
textdomain (PACKAGE);
#endif
// fprintf(stderr, "tesseract %s\n", tesseract::TessBaseAPI::Version());
// Make the order of args a bit more forgiving than it used to be.
const char* lang = "eng";
tesseract::PageSegMode pagesegmode = tesseract::PSM_SINGLE_BLOCK;
if (language!=NULL && language[0]!='\0')
lang = language;
/*
if (output == NULL)
{
fprintf(stderr, _("Usage:%s imagename outputbase [-l lang] "
"[-psm pagesegmode] [configfile...]\n"), argv[0]);
fprintf(stderr,
_("pagesegmode values are:\n"
"0 = Orientation and script detection (OSD) only.\n"
"1 = Automatic page segmentation with OSD.\n"
"2 = Automatic page segmentation, but no OSD, or OCR\n"
"3 = Fully automatic page segmentation, but no OSD. (Default)\n"
"4 = Assume a single column of text of variable sizes.\n"
"5 = Assume a single uniform block of vertically aligned text.\n"
"6 = Assume a single uniform block of text.\n"
"7 = Treat the image as a single text line.\n"
"8 = Treat the image as a single word.\n"
"9 = Treat the image as a single word in a circle.\n"
"10 = Treat the image as a single character.\n"));
fprintf(stderr, _("-l lang and/or -psm pagesegmode must occur before any"
"configfile.\n"));
exit(1);
}
*/
api.SetOutputName(NULL);
status=api.Init(datapath,lang,
ocr_type==0 ? tesseract::OEM_DEFAULT :
(ocr_type==1 ? tesseract::OEM_TESSERACT_ONLY :
(ocr_type==2 ? tesseract::OEM_CUBE_ONLY :
(tesseract::OEM_TESSERACT_CUBE_COMBINED))));
if (status)
return(status);
/*
api.Init("tesscapi",lang,tesseract::OEM_DEFAULT,
&(argv[arg]), argc - arg, NULL, NULL, false);
*/
// We have 2 possible sources of pagesegmode: a config file and
// the command line. For backwards compatability reasons, the
// default in tesseract is tesseract::PSM_SINGLE_BLOCK, but the
// default for this program is tesseract::PSM_AUTO. We will let
// the config file take priority, so the command-line default
// can take priority over the tesseract default, so we use the
// value from the command line only if the retrieved mode
// is still tesseract::PSM_SINGLE_BLOCK, indicating no change
// in any config file. Therefore the only way to force
// tesseract::PSM_SINGLE_BLOCK is from the command line.
// It would be simpler if we could set the value before Init,
// but that doesn't work.
if (api.GetPageSegMode() == tesseract::PSM_SINGLE_BLOCK)
api.SetPageSegMode(pagesegmode);
/*
** Initialization message
*/
{
char istr[256];
sprintf(istr,"Tesseract Open Source OCR Engine v%s ",tesseract::TessBaseAPI::Version());
if (ocr_type==0 || ocr_type==3)
sprintf(&istr[strlen(istr)],"[CUBE+] (lang=");
else if (ocr_type==2)
sprintf(&istr[strlen(istr)],"[CUBE] (lang=");
strncpy(&istr[strlen(istr)],language,253-strlen(istr));
istr[253]='\0';
strcat(istr,")");
if (out!=NULL)
fprintf(out,"%s\n",istr);
if (initstr!=NULL)
{
strncpy(initstr,istr,maxlen-1);
initstr[maxlen-1]='\0';
}
}
/* Turn off CUBE debugging output */
api.SetVariable("cube_debug_level","0");
#if (WILLUSDEBUG & 1)
api.SetVariable("cube_debug_level","9");
api.SetVariable("paragraph_debug_level","9");
api.SetVariable("tessdata_manager_debug_level","9");
api.SetVariable("tosp_debug_level","9");
api.SetVariable("wordrec_debug_level","9");
api.SetVariable("segsearch_debug_level","9");
#endif
return(0);
}