mirror of
https://github.com/tesseract-ocr/tesseract.git
synced 2024-11-27 12:49:35 +08:00
Remove scripts for training
They were replaced by Python3 scripts (part of the tesstrain repository). Signed-off-by: Stefan Weil <sw@weilnetz.de>
This commit is contained in:
Stefan Weil
parent
5eb2e86635
commit
93348a83a3
@ -795,10 +795,6 @@ training_CPPFLAGS += $(ICU_UC_CFLAGS) $(ICU_I18N_CFLAGS)
|
||||
training_CPPFLAGS += $(pango_CFLAGS)
|
||||
training_CPPFLAGS += $(cairo_CFLAGS)
|
||||
|
||||
bin_SCRIPTS = src/training/language-specific.sh src/training/tesstrain.sh
|
||||
scripts_DATA = src/training/tesstrain_utils.sh
|
||||
scriptsdir = $(bindir)
|
||||
|
||||
if DISABLED_LEGACY_ENGINE
|
||||
training_CPPFLAGS += -DDISABLED_LEGACY_ENGINE
|
||||
endif
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@ -1,98 +0,0 @@
|
||||
#!/bin/bash
|
||||
# (C) Copyright 2014, Google Inc.
|
||||
# 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.
|
||||
#
|
||||
# This script provides an easy way to execute various phases of training
|
||||
# Tesseract. For a detailed description of the phases, see
|
||||
# https://tesseract-ocr.github.io/tessdoc/Training-Tesseract.html.
|
||||
#
|
||||
|
||||
display_usage() {
|
||||
echo -e 'USAGE: tesstrain.sh
|
||||
--exposures EXPOSURES # A list of exposure levels to use (e.g. "-1 0 1").
|
||||
--fontlist FONTS # A list of fontnames to train on.
|
||||
--fonts_dir FONTS_PATH # Path to font files.
|
||||
--lang LANG_CODE # ISO 639 code.
|
||||
--langdata_dir DATADIR # Path to tesseract/training/langdata directory.
|
||||
--linedata_only # Only generate training data for lstmtraining.
|
||||
--output_dir OUTPUTDIR # Location of output traineddata file.
|
||||
--overwrite # Safe to overwrite files in output_dir.
|
||||
--run_shape_clustering # Run shape clustering (use for Indic langs).
|
||||
--maxpages # Specify maximum pages to output (default:0=all)
|
||||
--save_box_tiff # Save box/tiff pairs along with lstmf files.
|
||||
--xsize # Specify width of output image (default:3600)
|
||||
|
||||
OPTIONAL flag for specifying directory with user specified box/tiff pairs.
|
||||
Files should be named similar to ${LANG_CODE}.${fontname}.exp${EXPOSURE}.box/tif
|
||||
--my_boxtiff_dir MY_BOXTIFF_DIR # Location of user specified box/tiff files.
|
||||
|
||||
OPTIONAL flags for input data. If unspecified we will look for them in
|
||||
the langdata_dir directory.
|
||||
--training_text TEXTFILE # Text to render and use for training.
|
||||
--wordlist WORDFILE # Word list for the language ordered by
|
||||
# decreasing frequency.
|
||||
OPTIONAL flag to specify location of existing traineddata files, required
|
||||
during feature extraction. If unspecified will use TESSDATA_PREFIX defined in
|
||||
the current environment.
|
||||
--tessdata_dir TESSDATADIR # Path to tesseract/tessdata directory.
|
||||
NOTE:
|
||||
The font names specified in --fontlist need to be recognizable by Pango using
|
||||
fontconfig. An easy way to list the canonical names of all fonts available on
|
||||
your system is to run text2image with --list_available_fonts and the
|
||||
appropriate --fonts_dir path.'
|
||||
}
|
||||
|
||||
source "$(dirname $0)/tesstrain_utils.sh"
|
||||
if [[ $# -eq 0 || "$1" == "--help" || "$1" == "-h" ]]; then
|
||||
display_usage
|
||||
exit 0
|
||||
fi
|
||||
if [ $# == 0 ]; then
|
||||
display_usage
|
||||
exit 1
|
||||
fi
|
||||
|
||||
ARGV=("$@")
|
||||
parse_flags
|
||||
|
||||
mkdir -p ${TRAINING_DIR}
|
||||
|
||||
if [[ ${MY_BOXTIFF_DIR} != "" ]]; then
|
||||
tlog "\n=== Copy existing box/tiff pairs from '${MY_BOXTIFF_DIR}'"
|
||||
cp ${MY_BOXTIFF_DIR}/*.box ${TRAINING_DIR} | true
|
||||
cp ${MY_BOXTIFF_DIR}/*.tif ${TRAINING_DIR} | true
|
||||
ls -l ${TRAINING_DIR}
|
||||
fi
|
||||
|
||||
tlog "\n=== Starting training for language '${LANG_CODE}'"
|
||||
|
||||
source "$(dirname $0)/language-specific.sh"
|
||||
set_lang_specific_parameters ${LANG_CODE}
|
||||
|
||||
initialize_fontconfig
|
||||
|
||||
phase_I_generate_image 8
|
||||
phase_UP_generate_unicharset
|
||||
if $LINEDATA; then
|
||||
phase_E_extract_features " lstm.train " 8 "lstmf"
|
||||
make__lstmdata
|
||||
tlog "\nCreated starter traineddata for LSTM training of language '${LANG_CODE}'\n"
|
||||
tlog "\nRun 'lstmtraining' command to continue LSTM training for language '${LANG_CODE}'\n"
|
||||
else
|
||||
phase_D_generate_dawg
|
||||
phase_E_extract_features "box.train" 8 "tr"
|
||||
phase_C_cluster_prototypes "${TRAINING_DIR}/${LANG_CODE}.normproto"
|
||||
phase_S_cluster_shapes
|
||||
phase_M_cluster_microfeatures
|
||||
phase_B_generate_ambiguities
|
||||
make__traineddata
|
||||
tlog "\nCompleted training for language '${LANG_CODE}'\n"
|
||||
fi
|
||||
@ -1,632 +0,0 @@
|
||||
#!/bin/bash
|
||||
# (C) Copyright 2014, Google Inc.
|
||||
# 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.
|
||||
#
|
||||
# This script defines functions that are used by tesstrain.sh
|
||||
# For a detailed description of the phases, see
|
||||
# https://tesseract-ocr.github.io/tessdoc/Training-Tesseract.html.
|
||||
#
|
||||
# USAGE: source tesstrain_utils.sh
|
||||
|
||||
if [ -n "$BASH_VERSION" ];then
|
||||
set -u # comment in case of "unbound variable" error or fix the code
|
||||
set -eo pipefail;
|
||||
else
|
||||
echo "Warning: you aren't running script in bash - expect problems..."
|
||||
fi
|
||||
|
||||
UNAME=$(uname -s | tr 'A-Z' 'a-z')
|
||||
|
||||
FONT_CONFIG_CACHE=$(mktemp -d -t font_tmp.XXXXXXXXXX)
|
||||
|
||||
if [[ ($UNAME == *darwin*) ]]; then
|
||||
FONTS_DIR="/Library/Fonts/"
|
||||
else
|
||||
FONTS_DIR="/usr/share/fonts/"
|
||||
fi
|
||||
|
||||
DISTORT_IMAGE=false
|
||||
EXTRACT_FONT_PROPERTIES=false
|
||||
LINEDATA=false
|
||||
MAX_PAGES=0
|
||||
MY_BOXTIFF_DIR=""
|
||||
OUTPUT_DIR="/tmp/tesstrain/tessdata"
|
||||
OVERWRITE=false
|
||||
RUN_SHAPE_CLUSTERING=false
|
||||
SAVE_BOX_TIFF=false
|
||||
WORKSPACE_DIR=$(mktemp -d)
|
||||
X_SIZE=3600
|
||||
PT_SIZE=12
|
||||
|
||||
# set TESSDATA_PREFIX as empty, if not defined in environment to avoid an unbound variable
|
||||
TESSDATA_PREFIX=${TESSDATA_PREFIX:-}
|
||||
|
||||
# Logging helper functions.
|
||||
tlog() {
|
||||
if test -z "${LOG_FILE:-}"; then
|
||||
echo -e $*
|
||||
else
|
||||
echo -e $* | tee -a ${LOG_FILE}
|
||||
fi
|
||||
}
|
||||
|
||||
err_exit() {
|
||||
if test -z "${LOG_FILE:-}"; then
|
||||
echo -e "ERROR: "$*
|
||||
else
|
||||
echo -e "ERROR: "$* | tee -a ${LOG_FILE}
|
||||
fi
|
||||
exit 1
|
||||
}
|
||||
|
||||
# Helper function to run a command and append its output to a log. Aborts early
|
||||
# if the program file is not found.
|
||||
# Usage: run_command CMD ARG1 ARG2...
|
||||
run_command() {
|
||||
local cmd
|
||||
cmd=$(which $1 || \
|
||||
for d in api training; do
|
||||
which $d/$1 && break
|
||||
done) || err_exit "'$1' not found"
|
||||
shift
|
||||
tlog "[$(date)] ${cmd} $@"
|
||||
if ! "${cmd}" "$@" 2>&1 | tee -a "${LOG_FILE}"; then
|
||||
err_exit "Program $(basename ${cmd}) failed. Abort. Command line: ${cmd} $@"
|
||||
fi
|
||||
}
|
||||
|
||||
# Check if all the given files exist, or exit otherwise.
|
||||
# Used to check required input files and produced output files in each phase.
|
||||
# Usage: check_file_readable FILE1 FILE2...
|
||||
check_file_readable() {
|
||||
for file in $@; do
|
||||
if [[ ! -r ${file} ]]; then
|
||||
err_exit "${file} does not exist or is not readable"
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
# Sets the named variable to given value. Aborts if the value is missing or
|
||||
# if it looks like a flag.
|
||||
# Usage: parse_value VAR_NAME VALUE
|
||||
parse_value() {
|
||||
local val="${2:-}"
|
||||
if [[ -z "$val" ]]; then
|
||||
err_exit "Missing value for variable $1"
|
||||
exit
|
||||
fi
|
||||
if [[ ${val:0:2} == "--" ]]; then
|
||||
err_exit "Invalid value $val passed for variable $1"
|
||||
exit
|
||||
fi
|
||||
eval $1=\"$val\"
|
||||
}
|
||||
|
||||
# Does simple command-line parsing and initialization.
|
||||
parse_flags() {
|
||||
local i=0
|
||||
while test $i -lt ${#ARGV[@]}; do
|
||||
local j=$((i+1))
|
||||
case ${ARGV[$i]} in
|
||||
--)
|
||||
break;;
|
||||
--fontlist)
|
||||
fn=0
|
||||
FONTS=""
|
||||
while test $j -lt ${#ARGV[@]}; do
|
||||
test -z "${ARGV[$j]}" && break
|
||||
test $(echo ${ARGV[$j]} | cut -c -2) = "--" && break
|
||||
FONTS[$fn]="${ARGV[$j]}"
|
||||
fn=$((fn+1))
|
||||
j=$((j+1))
|
||||
done
|
||||
i=$((j-1)) ;;
|
||||
--exposures)
|
||||
exp=""
|
||||
while test $j -lt ${#ARGV[@]}; do
|
||||
test -z "${ARGV[$j]}" && break
|
||||
test $(echo ${ARGV[$j]} | cut -c -2) = "--" && break
|
||||
exp="$exp ${ARGV[$j]}"
|
||||
j=$((j+1))
|
||||
done
|
||||
parse_value "EXPOSURES" "$exp"
|
||||
i=$((j-1)) ;;
|
||||
--fonts_dir)
|
||||
parse_value "FONTS_DIR" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--tmp_dir)
|
||||
parse_value "TMP_DIR" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--lang)
|
||||
parse_value "LANG_CODE" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--langdata_dir)
|
||||
parse_value "LANGDATA_ROOT" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--maxpages)
|
||||
parse_value "MAX_PAGES" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--ptsize)
|
||||
parse_value "PT_SIZE" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--my_boxtiff_dir)
|
||||
parse_value "MY_BOXTIFF_DIR" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--distort_image)
|
||||
DISTORT_IMAGE=true ;;
|
||||
--output_dir)
|
||||
parse_value "OUTPUT_DIR" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--overwrite)
|
||||
OVERWRITE=true ;;
|
||||
--save_box_tiff)
|
||||
SAVE_BOX_TIFF=true ;;
|
||||
--linedata_only)
|
||||
LINEDATA=true ;;
|
||||
--extract_font_properties)
|
||||
EXTRACT_FONT_PROPERTIES=true ;;
|
||||
--noextract_font_properties)
|
||||
EXTRACT_FONT_PROPERTIES=false ;;
|
||||
--tessdata_dir)
|
||||
parse_value "TESSDATA_DIR" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--training_text)
|
||||
parse_value "TRAINING_TEXT" "${ARGV[$j]:-}"
|
||||
i=$j ;;
|
||||
--wordlist)
|
||||
parse_value "WORDLIST_FILE" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
--workspace_dir)
|
||||
rmdir "$FONT_CONFIG_CACHE"
|
||||
rmdir "$WORKSPACE_DIR"
|
||||
parse_value "WORKSPACE_DIR" ${ARGV[$j]:-}
|
||||
FONT_CONFIG_CACHE=$WORKSPACE_DIR/fc-cache
|
||||
mkdir -p $FONT_CONFIG_CACHE
|
||||
i=$j ;;
|
||||
--xsize)
|
||||
parse_value "X_SIZE" ${ARGV[$j]:-}
|
||||
i=$j ;;
|
||||
*)
|
||||
err_exit "Unrecognized argument ${ARGV[$i]}" ;;
|
||||
esac
|
||||
i=$((i+1))
|
||||
done
|
||||
if [[ -z ${LANG_CODE:-} ]]; then
|
||||
err_exit "Need to specify a language --lang"
|
||||
fi
|
||||
if [[ -z ${LANGDATA_ROOT:-} ]]; then
|
||||
err_exit "Need to specify path to language files --langdata_dir"
|
||||
fi
|
||||
if [[ -z ${TESSDATA_DIR:-} ]]; then
|
||||
if [[ -z ${TESSDATA_PREFIX} ]]; then
|
||||
err_exit "Need to specify a --tessdata_dir or have a "\
|
||||
"TESSDATA_PREFIX variable defined in your environment"
|
||||
else
|
||||
TESSDATA_DIR="${TESSDATA_PREFIX}"
|
||||
fi
|
||||
fi
|
||||
if [[ ! -d "${OUTPUT_DIR}" ]]; then
|
||||
tlog "Creating new directory ${OUTPUT_DIR}"
|
||||
mkdir -p "${OUTPUT_DIR}"
|
||||
fi
|
||||
|
||||
# Location where intermediate files will be created.
|
||||
TIMESTAMP=$(date +%Y-%m-%d)
|
||||
if [[ -z ${TMP_DIR:-} ]]; then
|
||||
TMP_DIR=$(mktemp -d -t ${LANG_CODE}-${TIMESTAMP}.XXX)
|
||||
else
|
||||
TMP_DIR=$(mktemp -d -p ${TMP_DIR} -t ${LANG_CODE}-${TIMESTAMP}.XXX)
|
||||
fi
|
||||
TRAINING_DIR=${TMP_DIR}
|
||||
# Location of log file for the whole run.
|
||||
LOG_FILE=${TRAINING_DIR}/tesstrain.log
|
||||
|
||||
# Take training text and wordlist from the langdata directory if not
|
||||
# specified in the command-line.
|
||||
TRAINING_TEXT=${TRAINING_TEXT:-${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.training_text}
|
||||
WORDLIST_FILE=${WORDLIST_FILE:-${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.wordlist}
|
||||
|
||||
WORD_BIGRAMS_FILE=${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.word.bigrams
|
||||
NUMBERS_FILE=${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.numbers
|
||||
PUNC_FILE=${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.punc
|
||||
BIGRAM_FREQS_FILE=${TRAINING_TEXT}.bigram_freqs
|
||||
UNIGRAM_FREQS_FILE=${TRAINING_TEXT}.unigram_freqs
|
||||
TRAIN_NGRAMS_FILE=${TRAINING_TEXT}.train_ngrams
|
||||
GENERATE_DAWGS=1
|
||||
}
|
||||
|
||||
# Function initializes font config with a unique font cache dir.
|
||||
initialize_fontconfig() {
|
||||
export FONT_CONFIG_CACHE
|
||||
local sample_path=${FONT_CONFIG_CACHE}/sample_text.txt
|
||||
echo "Text" >${sample_path}
|
||||
run_command text2image --fonts_dir=${FONTS_DIR} --ptsize ${PT_SIZE} \
|
||||
--font="${FONTS[0]}" --outputbase=${sample_path} --text=${sample_path} \
|
||||
--fontconfig_tmpdir=${FONT_CONFIG_CACHE}
|
||||
}
|
||||
|
||||
# Helper function for phaseI_generate_image. Generates the image for a single
|
||||
# language/font combination in a way that can be run in parallel.
|
||||
generate_font_image() {
|
||||
local font="$1"
|
||||
tlog "Rendering using ${font}"
|
||||
local fontname=$(echo ${font} | tr ' ' '_' | sed 's/,//g')
|
||||
local outbase=${TRAINING_DIR}/${LANG_CODE}.${fontname}.exp${EXPOSURE}
|
||||
|
||||
local common_args="--fontconfig_tmpdir=${FONT_CONFIG_CACHE}"
|
||||
common_args+=" --fonts_dir=${FONTS_DIR} --strip_unrenderable_words"
|
||||
common_args+=" --leading=${LEADING} --xsize=${X_SIZE}"
|
||||
common_args+=" --char_spacing=${CHAR_SPACING} --exposure=${EXPOSURE}"
|
||||
common_args+=" --outputbase=${outbase} --max_pages=${MAX_PAGES}"
|
||||
if $DISTORT_IMAGE; then
|
||||
common_args+=" --distort_image --invert=false"
|
||||
fi
|
||||
|
||||
# add --writing_mode=vertical-upright to common_args if the font is
|
||||
# specified to be rendered vertically.
|
||||
for vfont in "${VERTICAL_FONTS[@]}"; do
|
||||
if [[ "${font}" == "${vfont}" ]]; then
|
||||
common_args+=" --writing_mode=vertical-upright "
|
||||
break
|
||||
fi
|
||||
done
|
||||
|
||||
run_command text2image ${common_args} --font="${font}" --ptsize ${PT_SIZE} \
|
||||
--text=${TRAINING_TEXT} ${TEXT2IMAGE_EXTRA_ARGS:-}
|
||||
check_file_readable ${outbase}.box ${outbase}.tif
|
||||
|
||||
if $EXTRACT_FONT_PROPERTIES &&
|
||||
[[ -r ${TRAIN_NGRAMS_FILE} ]]; then
|
||||
tlog "Extracting font properties of ${font}"
|
||||
run_command text2image ${common_args} --font="${font}" \
|
||||
--ligatures=false --text=${TRAIN_NGRAMS_FILE} \
|
||||
--only_extract_font_properties --ptsize=32
|
||||
check_file_readable ${outbase}.fontinfo
|
||||
fi
|
||||
}
|
||||
|
||||
# Phase I : Generate (I)mages from training text for each font.
|
||||
phase_I_generate_image() {
|
||||
local par_factor=${1:-}
|
||||
if ! [[ "${par_factor}" -gt 0 ]]; then
|
||||
par_factor=1
|
||||
fi
|
||||
tlog "\n=== Phase I: Generating training images ==="
|
||||
if [[ -z ${TRAINING_TEXT:-} ]] || test ! -r "${TRAINING_TEXT}"; then
|
||||
err_exit "Could not find training text file ${TRAINING_TEXT:-}"
|
||||
fi
|
||||
CHAR_SPACING="0.0"
|
||||
|
||||
for EXPOSURE in $EXPOSURES; do
|
||||
if $EXTRACT_FONT_PROPERTIES && [[ -r ${BIGRAM_FREQS_FILE} ]]; then
|
||||
# Parse .bigram_freqs file and compose a .train_ngrams file with text
|
||||
# for tesseract to recognize during training. Take only the ngrams whose
|
||||
# combined weight accounts for 95% of all the bigrams in the language.
|
||||
NGRAM_FRAC=$(cat ${BIGRAM_FREQS_FILE} \
|
||||
| awk '{s=s+$2}; END {print (s/100)*p}' p=99)
|
||||
sort -rnk2 ${BIGRAM_FREQS_FILE} \
|
||||
| awk '{s=s+$2; if (s <= x) {printf "%s ", $1; } }' \
|
||||
x=${NGRAM_FRAC} > ${TRAIN_NGRAMS_FILE}
|
||||
check_file_readable ${TRAIN_NGRAMS_FILE}
|
||||
fi
|
||||
|
||||
local jobs=
|
||||
trap "kill $$" INT
|
||||
for font in "${FONTS[@]}"; do
|
||||
sleep 1
|
||||
test $(jobs -r | wc -l) -ge $par_factor && wait -n
|
||||
generate_font_image "${font}" &
|
||||
jobs="$jobs $!"
|
||||
done
|
||||
wait $jobs
|
||||
# Check that each process was successful.
|
||||
for font in "${FONTS[@]}"; do
|
||||
local fontname=$(echo ${font} | tr ' ' '_' | sed 's/,//g')
|
||||
local outbase=${TRAINING_DIR}/${LANG_CODE}.${fontname}.exp${EXPOSURE}
|
||||
check_file_readable ${outbase}.box ${outbase}.tif
|
||||
done
|
||||
done
|
||||
if $SAVE_BOX_TIFF && ( ! $LINEDATA ) ; then
|
||||
tlog "\n=== Saving box/tiff pairs for training data ==="
|
||||
for f in "${TRAINING_DIR}/${LANG_CODE}".*.box; do
|
||||
tlog "Moving ${f} to ${OUTPUT_DIR}"
|
||||
cp "${f}" "${OUTPUT_DIR}"
|
||||
done
|
||||
for f in "${TRAINING_DIR}/${LANG_CODE}".*.tif; do
|
||||
tlog "Moving ${f} to ${OUTPUT_DIR}"
|
||||
cp "${f}" "${OUTPUT_DIR}"
|
||||
done
|
||||
fi
|
||||
}
|
||||
|
||||
# Phase UP : Generate (U)nicharset and (P)roperties file.
|
||||
phase_UP_generate_unicharset() {
|
||||
tlog "\n=== Phase UP: Generating unicharset and unichar properties files ==="
|
||||
|
||||
local box_files=$(ls ${TRAINING_DIR}/*.box)
|
||||
UNICHARSET_FILE="${TRAINING_DIR}/${LANG_CODE}.unicharset"
|
||||
if [[ "${NORM_MODE}" == "2" ]] && [[ "${LANG_IS_RTL}" == "0" ]] ; then
|
||||
run_command unicharset_extractor --output_unicharset "${UNICHARSET_FILE}" \
|
||||
--norm_mode "${NORM_MODE}" ${TRAINING_TEXT}
|
||||
else
|
||||
run_command unicharset_extractor --output_unicharset "${UNICHARSET_FILE}" \
|
||||
--norm_mode "${NORM_MODE}" ${box_files}
|
||||
fi
|
||||
check_file_readable ${UNICHARSET_FILE}
|
||||
|
||||
XHEIGHTS_FILE="${TRAINING_DIR}/${LANG_CODE}.xheights"
|
||||
run_command set_unicharset_properties \
|
||||
-U ${UNICHARSET_FILE} -O ${UNICHARSET_FILE} -X ${XHEIGHTS_FILE} \
|
||||
--script_dir=${LANGDATA_ROOT}
|
||||
check_file_readable ${XHEIGHTS_FILE}
|
||||
}
|
||||
|
||||
# Phase D : Generate (D)awg files from unicharset file and wordlist files
|
||||
phase_D_generate_dawg() {
|
||||
tlog "\n=== Phase D: Generating Dawg files ==="
|
||||
|
||||
# Skip if requested
|
||||
if [[ ${GENERATE_DAWGS} -eq 0 ]]; then
|
||||
tlog "Skipping ${phase_name}"
|
||||
return
|
||||
fi
|
||||
|
||||
# Output files
|
||||
WORD_DAWG=${TRAINING_DIR}/${LANG_CODE}.word-dawg
|
||||
FREQ_DAWG=${TRAINING_DIR}/${LANG_CODE}.freq-dawg
|
||||
PUNC_DAWG=${TRAINING_DIR}/${LANG_CODE}.punc-dawg
|
||||
NUMBER_DAWG=${TRAINING_DIR}/${LANG_CODE}.number-dawg
|
||||
BIGRAM_DAWG=${TRAINING_DIR}/${LANG_CODE}.bigram-dawg
|
||||
|
||||
# Word DAWG
|
||||
local freq_wordlist_file=${TRAINING_DIR}/${LANG_CODE}.wordlist.clean.freq
|
||||
if [[ -s ${WORDLIST_FILE} ]]; then
|
||||
tlog "Generating word Dawg"
|
||||
check_file_readable ${UNICHARSET_FILE}
|
||||
run_command wordlist2dawg -r 1 ${WORDLIST_FILE} ${WORD_DAWG} \
|
||||
${UNICHARSET_FILE}
|
||||
check_file_readable ${WORD_DAWG}
|
||||
|
||||
FREQ_DAWG_SIZE=100
|
||||
head -n ${FREQ_DAWG_SIZE} ${WORDLIST_FILE} > ${freq_wordlist_file}
|
||||
fi
|
||||
|
||||
# Freq-word DAWG
|
||||
if [[ -s ${freq_wordlist_file} ]]; then
|
||||
check_file_readable ${UNICHARSET_FILE}
|
||||
tlog "Generating frequent-word Dawg"
|
||||
run_command wordlist2dawg -r 1 ${freq_wordlist_file} \
|
||||
${FREQ_DAWG} ${UNICHARSET_FILE}
|
||||
check_file_readable ${FREQ_DAWG}
|
||||
fi
|
||||
|
||||
# Punctuation DAWG
|
||||
# -r arguments to wordlist2dawg denote RTL reverse policy
|
||||
# (see Trie::RTLReversePolicy enum in tesseract/src/dict/trie.h).
|
||||
# We specify 0/RRP_DO_NO_REVERSE when generating number DAWG,
|
||||
# 1/RRP_REVERSE_IF_HAS_RTL for freq and word DAWGS,
|
||||
# 2/RRP_FORCE_REVERSE for the punctuation DAWG.
|
||||
local punc_reverse_policy=0;
|
||||
if [[ "${LANG_IS_RTL}" == "1" ]]; then
|
||||
punc_reverse_policy=2
|
||||
fi
|
||||
if [[ ! -s ${PUNC_FILE} ]]; then
|
||||
PUNC_FILE="${LANGDATA_ROOT}/common.punc"
|
||||
fi
|
||||
check_file_readable ${PUNC_FILE}
|
||||
run_command wordlist2dawg -r ${punc_reverse_policy} \
|
||||
${PUNC_FILE} ${PUNC_DAWG} ${UNICHARSET_FILE}
|
||||
check_file_readable ${PUNC_DAWG}
|
||||
|
||||
# Numbers DAWG
|
||||
if [[ -s ${NUMBERS_FILE} ]]; then
|
||||
run_command wordlist2dawg -r 0 \
|
||||
${NUMBERS_FILE} ${NUMBER_DAWG} ${UNICHARSET_FILE}
|
||||
check_file_readable ${NUMBER_DAWG}
|
||||
fi
|
||||
|
||||
# Bigram dawg
|
||||
if [[ -s ${WORD_BIGRAMS_FILE} ]]; then
|
||||
run_command wordlist2dawg -r 1 \
|
||||
${WORD_BIGRAMS_FILE} ${BIGRAM_DAWG} ${UNICHARSET_FILE}
|
||||
check_file_readable ${BIGRAM_DAWG}
|
||||
fi
|
||||
}
|
||||
|
||||
# Phase E : (E)xtract .tr feature files from .tif/.box files
|
||||
phase_E_extract_features() {
|
||||
local box_config=$1
|
||||
local par_factor=$2
|
||||
local ext=$3
|
||||
if ! [[ "${par_factor}" -gt 0 ]]; then
|
||||
par_factor=1
|
||||
fi
|
||||
tlog "\n=== Phase E: Generating ${ext} files ==="
|
||||
|
||||
local img_files=""
|
||||
for exposure in ${EXPOSURES}; do
|
||||
img_files=${img_files}' '$(ls ${TRAINING_DIR}/*.exp${exposure}.tif)
|
||||
done
|
||||
|
||||
# Use any available language-specific configs.
|
||||
local config=""
|
||||
if [[ -r ${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.config ]]; then
|
||||
config=${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.config
|
||||
fi
|
||||
|
||||
OLD_TESSDATA_PREFIX=${TESSDATA_PREFIX}
|
||||
export TESSDATA_PREFIX=${TESSDATA_DIR}
|
||||
tlog "Using TESSDATA_PREFIX=${TESSDATA_PREFIX}"
|
||||
local jobs=
|
||||
trap "kill $$" INT
|
||||
for img_file in ${img_files}; do
|
||||
test $(jobs -r | wc -l) -ge $par_factor && wait -n
|
||||
run_command tesseract ${img_file} ${img_file%.*} \
|
||||
${box_config} ${config} &
|
||||
jobs="$jobs $!"
|
||||
done
|
||||
wait $jobs
|
||||
export TESSDATA_PREFIX=${OLD_TESSDATA_PREFIX}
|
||||
# Check that all the output files were produced.
|
||||
for img_file in ${img_files}; do
|
||||
check_file_readable "${img_file%.*}.${ext}"
|
||||
done
|
||||
}
|
||||
|
||||
# Phase C : (C)luster feature prototypes in .tr into normproto file (cnTraining)
|
||||
# phaseC_cluster_prototypes ${TRAINING_DIR}/${LANG_CODE}.normproto
|
||||
phase_C_cluster_prototypes() {
|
||||
tlog "\n=== Phase C: Clustering feature prototypes (cnTraining) ==="
|
||||
local out_normproto=$1
|
||||
|
||||
run_command cntraining -D "${TRAINING_DIR}/" \
|
||||
$(ls ${TRAINING_DIR}/*.tr)
|
||||
|
||||
check_file_readable ${TRAINING_DIR}/normproto
|
||||
mv ${TRAINING_DIR}/normproto ${out_normproto}
|
||||
}
|
||||
|
||||
# Phase S : (S)hape clustering
|
||||
phase_S_cluster_shapes() {
|
||||
if ! $RUN_SHAPE_CLUSTERING; then
|
||||
tlog "\n=== Shape Clustering disabled ==="
|
||||
return
|
||||
fi
|
||||
check_file_readable ${LANGDATA_ROOT}/font_properties
|
||||
local font_props="-F ${LANGDATA_ROOT}/font_properties"
|
||||
if [[ -r ${TRAINING_DIR}/${LANG_CODE}.xheights ]] &&\
|
||||
[[ -s ${TRAINING_DIR}/${LANG_CODE}.xheights ]]; then
|
||||
font_props=${font_props}" -X ${TRAINING_DIR}/${LANG_CODE}.xheights"
|
||||
fi
|
||||
|
||||
run_command shapeclustering \
|
||||
-D "${TRAINING_DIR}/" \
|
||||
-U ${TRAINING_DIR}/${LANG_CODE}.unicharset \
|
||||
-O ${TRAINING_DIR}/${LANG_CODE}.mfunicharset \
|
||||
${font_props} \
|
||||
$(ls ${TRAINING_DIR}/*.tr)
|
||||
check_file_readable ${TRAINING_DIR}/shapetable \
|
||||
${TRAINING_DIR}/${LANG_CODE}.mfunicharset
|
||||
}
|
||||
|
||||
# Phase M : Clustering microfeatures (mfTraining)
|
||||
phase_M_cluster_microfeatures() {
|
||||
tlog "\n=== Phase M : Clustering microfeatures (mfTraining) ==="
|
||||
|
||||
check_file_readable ${LANGDATA_ROOT}/font_properties
|
||||
font_props="-F ${LANGDATA_ROOT}/font_properties"
|
||||
if [[ -r ${TRAINING_DIR}/${LANG_CODE}.xheights ]] && \
|
||||
[[ -s ${TRAINING_DIR}/${LANG_CODE}.xheights ]]; then
|
||||
font_props=${font_props}" -X ${TRAINING_DIR}/${LANG_CODE}.xheights"
|
||||
fi
|
||||
|
||||
run_command mftraining \
|
||||
-D "${TRAINING_DIR}/" \
|
||||
-U ${TRAINING_DIR}/${LANG_CODE}.unicharset \
|
||||
-O ${TRAINING_DIR}/${LANG_CODE}.mfunicharset \
|
||||
${font_props} \
|
||||
$(ls ${TRAINING_DIR}/*.tr)
|
||||
check_file_readable ${TRAINING_DIR}/inttemp ${TRAINING_DIR}/shapetable \
|
||||
${TRAINING_DIR}/pffmtable ${TRAINING_DIR}/${LANG_CODE}.mfunicharset
|
||||
mv ${TRAINING_DIR}/inttemp ${TRAINING_DIR}/${LANG_CODE}.inttemp
|
||||
mv ${TRAINING_DIR}/shapetable ${TRAINING_DIR}/${LANG_CODE}.shapetable
|
||||
mv ${TRAINING_DIR}/pffmtable ${TRAINING_DIR}/${LANG_CODE}.pffmtable
|
||||
mv ${TRAINING_DIR}/${LANG_CODE}.mfunicharset ${TRAINING_DIR}/${LANG_CODE}.unicharset
|
||||
}
|
||||
|
||||
phase_B_generate_ambiguities() {
|
||||
tlog "\n=== Phase B : ambiguities training ==="
|
||||
|
||||
# Check for manually created ambiguities data.
|
||||
if [[ -r ${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.unicharambigs ]]; then
|
||||
tlog "Found file ${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.unicharambigs"
|
||||
cp ${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}.unicharambigs \
|
||||
${TRAINING_DIR}/${LANG_CODE}.unicharambigs
|
||||
# Make it writable, as it may be read-only in the client.
|
||||
chmod u+w ${TRAINING_DIR}/${LANG_CODE}.unicharambigs
|
||||
return
|
||||
else
|
||||
tlog "No unicharambigs file found!"
|
||||
fi
|
||||
|
||||
# TODO: Add support for generating ambiguities automatically.
|
||||
}
|
||||
|
||||
make__lstmdata() {
|
||||
tlog "\n=== Constructing LSTM training data ==="
|
||||
local lang_prefix="${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}"
|
||||
local lang_is_rtl=""
|
||||
if [[ "${LANG_IS_RTL}" == "1" ]]; then
|
||||
lang_is_rtl="--lang_is_rtl"
|
||||
fi
|
||||
local pass_through=""
|
||||
if [[ "${NORM_MODE}" -ge "2" ]]; then
|
||||
pass_through="--pass_through_recoder"
|
||||
fi
|
||||
|
||||
# Build the starter traineddata from the inputs.
|
||||
run_command combine_lang_model \
|
||||
--input_unicharset "${TRAINING_DIR}/${LANG_CODE}.unicharset" \
|
||||
--script_dir "${LANGDATA_ROOT}" \
|
||||
--words "${lang_prefix}.wordlist" \
|
||||
--numbers "${lang_prefix}.numbers" \
|
||||
--puncs "${lang_prefix}.punc" \
|
||||
--output_dir "${OUTPUT_DIR}" --lang "${LANG_CODE}" \
|
||||
"${pass_through}" "${lang_is_rtl}"
|
||||
|
||||
if $SAVE_BOX_TIFF; then
|
||||
tlog "\n=== Saving box/tiff pairs for training data ==="
|
||||
for f in "${TRAINING_DIR}/${LANG_CODE}".*.box; do
|
||||
tlog "Moving ${f} to ${OUTPUT_DIR}"
|
||||
mv "${f}" "${OUTPUT_DIR}"
|
||||
done
|
||||
for f in "${TRAINING_DIR}/${LANG_CODE}".*.tif; do
|
||||
tlog "Moving ${f} to ${OUTPUT_DIR}"
|
||||
mv "${f}" "${OUTPUT_DIR}"
|
||||
done
|
||||
fi
|
||||
|
||||
tlog "\n=== Moving lstmf files for training data ==="
|
||||
for f in "${TRAINING_DIR}/${LANG_CODE}".*.lstmf; do
|
||||
tlog "Moving ${f} to ${OUTPUT_DIR}"
|
||||
mv "${f}" "${OUTPUT_DIR}"
|
||||
done
|
||||
local lstm_list="${OUTPUT_DIR}/${LANG_CODE}.training_files.txt"
|
||||
ls -1 "${OUTPUT_DIR}/${LANG_CODE}".*.lstmf > "${lstm_list}"
|
||||
}
|
||||
|
||||
make__traineddata() {
|
||||
tlog "\n=== Making final traineddata file ==="
|
||||
local lang_prefix=${LANGDATA_ROOT}/${LANG_CODE}/${LANG_CODE}
|
||||
|
||||
# Combine available files for this language from the langdata dir.
|
||||
if [[ -r ${lang_prefix}.config ]]; then
|
||||
tlog "Copying ${lang_prefix}.config to ${TRAINING_DIR}"
|
||||
cp ${lang_prefix}.config ${TRAINING_DIR}
|
||||
chmod u+w ${TRAINING_DIR}/${LANG_CODE}.config
|
||||
fi
|
||||
if [[ -r ${lang_prefix}.params-model ]]; then
|
||||
tlog "Copying ${lang_prefix}.params-model to ${TRAINING_DIR}"
|
||||
cp ${lang_prefix}.params-model ${TRAINING_DIR}
|
||||
chmod u+w ${TRAINING_DIR}/${LANG_CODE}.params-model
|
||||
fi
|
||||
|
||||
# Compose the traineddata file.
|
||||
run_command combine_tessdata ${TRAINING_DIR}/${LANG_CODE}.
|
||||
|
||||
# Copy it to the output dir, overwriting only if allowed by the cmdline flag.
|
||||
local destfile=${OUTPUT_DIR}/${LANG_CODE}.traineddata;
|
||||
if [[ -f ${destfile} ]] && ! $OVERWRITE; then
|
||||
err_exit "File ${destfile} exists and no --overwrite specified";
|
||||
fi
|
||||
tlog "Moving ${TRAINING_DIR}/${LANG_CODE}.traineddata to ${OUTPUT_DIR}"
|
||||
cp -f ${TRAINING_DIR}/${LANG_CODE}.traineddata ${destfile}
|
||||
}
|
||||
@ -35,7 +35,7 @@ public:
|
||||
// Loads a set of lstmf files that were created using the lstm.train config to
|
||||
// tesseract into memory ready for testing. Returns false if nothing was
|
||||
// loaded. The arg is a filename of a file that lists the filenames, with one
|
||||
// name per line. Conveniently, tesstrain.sh generates such a file, along
|
||||
// name per line. Conveniently, tesstrain.py generates such a file, along
|
||||
// with the files themselves.
|
||||
bool LoadAllEvalData(const char *filenames_file);
|
||||
// Loads a set of lstmf files that were created using the lstm.train config to
|
||||
|
||||
@ -15,7 +15,7 @@
|
||||
//
|
||||
// Use --xsize 800 for text2image to be similar to original training data.
|
||||
//
|
||||
// src/training/tesstrain.sh --fonts_dir /usr/share/fonts --lang eng \
|
||||
// tesstrain.py --fonts_dir /usr/share/fonts --lang eng \
|
||||
// --linedata_only --noextract_font_properties --langdata_dir ../langdata_lstm \
|
||||
// --tessdata_dir ../tessdata --output_dir ~/tesseract/test/testdata \
|
||||
// --fontlist "Arial" --maxpages 10
|
||||
|
||||
Loading…
Reference in New Issue
Block a user