Skip to content

Commit

Permalink
test ocropus-rtrain --file/-f feature, help message
Browse files Browse the repository at this point in the history
  • Loading branch information
kba committed Dec 12, 2017
1 parent ba9b573 commit 7bbc37c
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 9 deletions.
2 changes: 1 addition & 1 deletion ocropus-rtrain
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ parser.add_argument("-Q","--nocheck",action="store_true")
parser.add_argument("-p","--pad",type=int,default=16)

# add file
parser.add_argument("-f","--file",default=None,help="input the training data as a file list")
parser.add_argument("-f","--file",default=None,help="path to file listing input files, one per line")

parser.add_argument("files",nargs="*")
args = parser.parse_args()
Expand Down
37 changes: 29 additions & 8 deletions run-test-ci
Original file line number Diff line number Diff line change
@@ -1,5 +1,12 @@
#!/bin/bash -e

# Usage: ./run-test-ci [tests...]
#
# Examples:
#
# ./run-test-ci # Run all tests
# ./run-test-ci page conf # Run only 'test_page' and 'test_conf'
#
BASE=$(dirname $0)

# 'RUNNER' is the binary that the scripts are executed by. It defaults to
Expand Down Expand Up @@ -61,6 +68,13 @@ test_rtrain() {
$RUNNER $BASE/ocropus-rtrain 'book/*/*.bin.png' -N 5 -o ci-test-model
}

test_rtrain_files() {
tar -zxf $BASE/tests/uw3-500.tgz
find 'book' -name '*.bin.png' > INPUT_FILES
$RUNNER $BASE/ocropus-rtrain -f INPUT_FILES -N 5 -o ci-test-model
rm INPUT_FILES
}

test_nlbin() {
local TESTIMAGE=0071-010012.png
cp $BASE/tests/$TESTIMAGE temp
Expand Down Expand Up @@ -100,11 +114,18 @@ test_gtedit() {

rm -rf temp
mkdir -p temp
test_page
test_conf
test_linegen
test_rtrain
test_nlbin
test_gpageseg
test_rpred
test_gtedit
if (( $# > 0 ));then
for test in "$@";do
test_$test
done
else
test_page
test_conf
test_linegen
test_rtrain
test_rtrain_files
test_nlbin
test_gpageseg
test_rpred
test_gtedit
fi

0 comments on commit 7bbc37c

Please sign in to comment.