summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--bucket-lines-prob/bucket-lines-prob.go (renamed from line-conf-buckets/line-conf-buckets.go)24
1 files changed, 13 insertions, 11 deletions
diff --git a/line-conf-buckets/line-conf-buckets.go b/bucket-lines-prob/bucket-lines-prob.go
index 689b779..728268d 100644
--- a/line-conf-buckets/line-conf-buckets.go
+++ b/bucket-lines-prob/bucket-lines-prob.go
@@ -1,8 +1,5 @@
package main
-// TODO: rename this emphasising that it's for .prob/ocropy
-// TODO: set bucket dirname from cmdline
-
import (
"flag"
"fmt"
@@ -14,12 +11,23 @@ import (
)
func main() {
+ b := parse.BucketSpecs{
+ // minimum confidence, name
+ { 0, "bad" },
+ { 0.95, "95to98" },
+ { 0.98, "98plus" },
+ }
+
flag.Usage = func() {
- fmt.Fprintf(os.Stderr, "Usage: line-conf-buckets prob1 [prob2] [...]\n")
+ fmt.Fprintf(os.Stderr, "Usage: bucket-lines-prob [-d dir] prob1 [prob2] [...]\n")
fmt.Fprintf(os.Stderr, "Copies image-text line pairs into different directories according\n")
fmt.Fprintf(os.Stderr, "to the average character probability for the line.\n")
+ fmt.Fprintf(os.Stderr, "This uses the .prob files generated by ocropy-rpred's --probabilities\n")
+ fmt.Fprintf(os.Stderr, "option, which it assumes will be in the same directory as the line's\n")
+ fmt.Fprintf(os.Stderr, "image and text files.\n")
flag.PrintDefaults()
}
+ dir := flag.String("d", "buckets", "Directory to store the buckets")
flag.Parse()
if flag.NArg() < 1 {
flag.Usage()
@@ -39,13 +47,7 @@ func main() {
}
}
- b := parse.BucketSpecs{
- { 0, "bad" },
- { 0.95, "95to98" },
- { 0.98, "98plus" },
- }
-
- stats, err := parse.BucketUp(lines, b, "newbuckets")
+ stats, err := parse.BucketUp(lines, b, *dir)
if err != nil {
log.Fatal(err)
}