You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.1 KiB
Bash
54 lines
1.1 KiB
Bash
#!/bin/sh
|
|
# Train a GT4HistOCR Calamari model
|
|
# (or rather 5 for voted prediction)
|
|
|
|
set -e
|
|
|
|
self=`realpath $0`
|
|
self_dir=`dirname "$self"`
|
|
|
|
|
|
|
|
|
|
cd $self_dir
|
|
DATA_SUBDIR=data
|
|
get_from_annex() {
|
|
annex_get 'GT4HistOCR/corpus/*.tar.bz2'
|
|
}
|
|
get_from_web() {
|
|
download_to 'https://zenodo.org/record/1344132/files/GT4HistOCR.tar?download=1' 'GT4HistOCR'
|
|
}
|
|
. $self_dir/qurator_data_lib.sh
|
|
handle_data
|
|
|
|
|
|
|
|
|
|
|
|
rm -rf /tmp/train-calamari-gt4histocr.*
|
|
TMPDIR=`mktemp -d /tmp/train-calamari-gt4histocr.XXXXX`
|
|
|
|
echo "Unpacking dataset tar files to $TMPDIR"
|
|
for tar in $DATA_SUBDIR/GT4HistOCR/corpus/*.tar.bz2; do
|
|
tar xf $tar -C $TMPDIR
|
|
done
|
|
echo "Removing dta19/1882-keller_sinngedicht/04970.nrm.png (Broken PNG)"
|
|
rm -f $TMPDIR/dta19/1882-keller_sinngedicht/04970.*
|
|
|
|
export PYTHONUNBUFFERED=1 # For python + tee
|
|
|
|
outdir=$DATA_SUBDIR/calamari-models/GT4HistOCR
|
|
mkdir -p $outdir
|
|
|
|
calamari-cross-fold-train \
|
|
--files \
|
|
"$TMPDIR/*/*/*.png" \
|
|
--best_models_dir $outdir \
|
|
--early_stopping_frequency=0.25 \
|
|
--early_stopping_nbest=5 \
|
|
--batch_size=128 \
|
|
--n_folds=5 \
|
|
--max_parallel_models=1 \
|
|
--display=0.01 \
|
|
2>&1 | tee $outdir/train.`date -Iminutes`.log
|