Rename ocropus bucket tool, add -d option, and improve documentation

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)
 	}