mirror of
https://github.com/tesseract-ocr/tesseract.git
synced 2024-12-12 23:49:06 +08:00
88 lines
4.3 KiB
C++
88 lines
4.3 KiB
C++
// Copyright 2017 Google Inc. All Rights Reserved.
|
|
// Author: rays@google.com (Ray Smith)
|
|
// Purpose: Program to generate a traineddata file that can be used to train an
|
|
// LSTM-based neural network model from a unicharset and an optional
|
|
// set of wordlists. Eliminates the need to run
|
|
// set_unicharset_properties, wordlist2dawg, some non-existent binary
|
|
// to generate the recoder, and finally combine_tessdata.
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
#include "commandlineflags.h"
|
|
#include "lang_model_helpers.h"
|
|
#include "tprintf.h"
|
|
#include "unicharset_training_utils.h"
|
|
|
|
STRING_PARAM_FLAG(input_unicharset, "",
|
|
"Unicharset to complete and use in encoding");
|
|
STRING_PARAM_FLAG(script_dir, "",
|
|
"Directory name for input script unicharsets");
|
|
STRING_PARAM_FLAG(words, "",
|
|
"File listing words to use for the system dictionary");
|
|
STRING_PARAM_FLAG(puncs, "", "File listing punctuation patterns");
|
|
STRING_PARAM_FLAG(numbers, "", "File listing number patterns");
|
|
STRING_PARAM_FLAG(output_dir, "", "Root directory for output files");
|
|
STRING_PARAM_FLAG(version_str, "", "Version string to add to traineddata file");
|
|
STRING_PARAM_FLAG(lang, "", "Name of language being processed");
|
|
BOOL_PARAM_FLAG(lang_is_rtl, false,
|
|
"True if lang being processed is written right-to-left");
|
|
BOOL_PARAM_FLAG(pass_through_recoder, false,
|
|
"If true, the recoder is a simple pass-through of the"
|
|
" unicharset. Otherwise, potentially a compression of it");
|
|
|
|
int main(int argc, char** argv) {
|
|
tesseract::ParseCommandLineFlags(argv[0], &argc, &argv, true);
|
|
|
|
// Check validity of input flags.
|
|
if (FLAGS_input_unicharset.empty() || FLAGS_script_dir.empty() ||
|
|
FLAGS_output_dir.empty() || FLAGS_lang.empty()) {
|
|
tprintf("Usage: %s --input_unicharset filename --script_dir dirname\n",
|
|
argv[0]);
|
|
tprintf(" --output_dir rootdir --lang lang [--lang_is_rtl]\n");
|
|
tprintf(" [--words file --puncs file --numbers file]\n");
|
|
tprintf("Sets properties on the input unicharset file, and writes:\n");
|
|
tprintf("rootdir/lang/lang.charset_size=ddd.txt\n");
|
|
tprintf("rootdir/lang/lang.traineddata\n");
|
|
tprintf("rootdir/lang/lang.unicharset\n");
|
|
tprintf("If the 3 word lists are provided, the dawgs are also added to");
|
|
tprintf(" the traineddata file.\n");
|
|
tprintf("The output unicharset and charset_size files are just for human");
|
|
tprintf(" readability.\n");
|
|
exit(1);
|
|
}
|
|
GenericVector<STRING> words, puncs, numbers;
|
|
// If these reads fail, we get a warning message and an empty list of words.
|
|
tesseract::ReadFile(FLAGS_words.c_str(), nullptr).split('\n', &words);
|
|
tesseract::ReadFile(FLAGS_puncs.c_str(), nullptr).split('\n', &puncs);
|
|
tesseract::ReadFile(FLAGS_numbers.c_str(), nullptr).split('\n', &numbers);
|
|
// Load the input unicharset
|
|
UNICHARSET unicharset;
|
|
if (!unicharset.load_from_file(FLAGS_input_unicharset.c_str(), false)) {
|
|
tprintf("Failed to load unicharset from %s\n",
|
|
FLAGS_input_unicharset.c_str());
|
|
return 1;
|
|
}
|
|
tprintf("Loaded unicharset of size %d from file %s\n", unicharset.size(),
|
|
FLAGS_input_unicharset.c_str());
|
|
|
|
// Set unichar properties
|
|
tprintf("Setting unichar properties\n");
|
|
tesseract::SetupBasicProperties(/*report_errors*/ true,
|
|
/*decompose (NFD)*/ false, &unicharset);
|
|
tprintf("Setting script properties\n");
|
|
tesseract::SetScriptProperties(FLAGS_script_dir.c_str(), &unicharset);
|
|
// Combine everything into a traineddata file.
|
|
return tesseract::CombineLangModel(
|
|
unicharset, FLAGS_script_dir.c_str(), FLAGS_version_str.c_str(),
|
|
FLAGS_output_dir.c_str(), FLAGS_lang.c_str(), FLAGS_pass_through_recoder,
|
|
words, puncs, numbers, FLAGS_lang_is_rtl, /*reader*/ nullptr,
|
|
/*writer*/ nullptr);
|
|
}
|