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.
ocrd-galley/build

95 lines
2.2 KiB
Bash

#!/bin/sh
DATA_SUBDIR=data
get_from_annex() {
annex_get 'calamari-models/GT4HistOCR/*.ckpt*'
annex_get 'tesseract-models/GT4HistOCR/*.traineddata'
annex_get 'textline_detection/*.h5'
}
get_from_web() {
download_to 'https://file.spk-berlin.de:8443/calamari-models/GT4HistOCR/model.tar.xz' 'calamari-models/GT4HistOCR'
download_to 'https://file.spk-berlin.de:8443/tesseract-models/GT4HistOCR/models.tar' 'tesseract-models/GT4HistOCR'
download_to 'https://file.spk-berlin.de:8443/textline_detection/models.tar.gz' 'textline_detection'
}
check_data_subdir() {
result=0
if git submodule status $DATA_SUBDIR | grep -q '^-'; then
echo "$DATA_SUBDIR/ is not an initialized submodule"; result=1
fi
if ! [ -e $DATA_SUBDIR/.git/annex ]; then
echo "$DATA_SUBDIR/ is not a git annex repository"; result=1
fi
if ! (cd $DATA_SUBDIR && git annex version | grep -q 'local repository version: 7'); then
echo "$DATA_SUBDIR/ is not a git annex repository version 7"; result=1
fi
if ! (cd $DATA_SUBDIR && git remote | grep -q '^nfs$'); then
echo "$DATA_SUBDIR/ has no git remote 'nfs'"; result=1
fi
return $result
}
suggest_commands() {
echo "Suggested commands:"
echo
echo "git submodule update --init"
echo "(cd $DATA_SUBDIR && git annex init --version=7)"
echo "(cd $DATA_SUBDIR && git remote add nfs /<... path to ...>/GitNX-Repository/qurator/qurator-data)"
}
annex_get() {
file_pattern="$1"
(
cd data
git annex get $file_pattern
# fsck seems to be necessary to fix the files if we're in a submodule
git annex fsck $file_pattern
)
}
download_to() {
download_source="$1"
unpack_to="$2"
(
cd data
tmpf=`mktemp 'tmp.XXXXX'`
wget -O $tmpf "$download_source"
mkdir -p "$unpack_to"
# XXX Unpacking relies on tar -a unpacking any tar compression, might not work everywhere?
tar -C "$unpack_to" -af $tmpf -xv
rm -f $tmpf
)
}
set -e
if ! check_data_subdir; then
select choice in "Abort to manually fix $DATA_SUBDIR submodule" "Download data files from the web"; do
if [ $REPLY = 1 ]; then
suggest_commands
exit
else
get_from_web
break
fi
done
else
get_from_annex
fi
docker build -t my_ocrd_workflow .