Added command line page recognizer.

ocropus3-recognize

@@ -0,0 +1,95 @@
+#!/usr/bin/env python
+import sys
+from pylab import *
+import ocroline
+import ocroseg
+import ocrorot # Loading the RotationEstimator Model
+import ocrobin
+import argparse
+from dlinputs import paths
+import scipy.ndimage as ndi
+
+import dltrainers.layers
+sys.modules["layers"] = dltrainers.layers
+
+model_path = ["."]
+for d in "ocrobin ocroseg ocrorot ocroline".split():
+    for p in "/usr/local/share /usr/share".split():
+        model_path += [p+"/"+d]
+model_path = ":".join(model_path)
+
+def findmodel(s):
+    result = paths.find_file(model_path, s)
+    assert result is not None, "{} not found on path".format(s)
+    return result
+
+parser = argparse.ArgumentParser("train a page segmenter")
+parser.add_argument("--binarizer", default="bin-000000046-005393.pt")
+parser.add_argument("--rotation", default="rot-000003456-020897.pt")
+parser.add_argument("--skew", default="logskew-000015808-000132.pt")
+parser.add_argument("--segmenter", default="lowskew-000000259-011440.pt")
+parser.add_argument("--linerec", default="line2-000003330-004377.pt")
+parser.add_argument("--display", type=float, default=-1.0)
+parser.add_argument("inputs", nargs="+")
+args = parser.parse_args()
+
+def note(*args):
+    args = [str(x) for x in args]
+    sys.stderr.write("# " + " ".join(args) + "\n")
+
+bm = rot = sk = seg = rec = None
+
+if args.binarizer != "none":
+    bm = ocrobin.Binarizer(findmodel(args.binarizer))
+if args.rotation != "none":
+    rot = ocrorot.RotationEstimator(findmodel(args.rotation))
+if args.skew != "none":
+    sk = ocrorot.SkewEstimator(findmodel(args.skew))
+if args.segmenter != "none":
+    seg = ocroseg.Segmenter(findmodel(args.segmenter))
+if args.linerec != "none":
+    rec = ocroline.LineRecognizer(findmodel(args.linerec))
+
+if args.display > 0:
+    ion()
+    rc("image", cmap="gray", interpolation="bicubic")
+
+def debug_display(image):
+    if args.display > 0:
+        clf()
+        imshow(image)
+        ginput(1, args.display)
+
+for fname in args.inputs:
+    image = imread(fname)
+    note(fname, image.shape, amin(image), amax(image), mean(image))
+    if image.ndim==3: image = mean(image[:, :, :3], 2)
+    debug_display(image)
+
+    if bm is not None:
+        image = bm.binarize(image)
+        note("binarized")
+        debug_display(image)
+
+    if rot is not None:
+        angle = rot.rotation(image)
+        note("rotation", angle)
+        if angle != 0.0:
+            image = ndi.rotate(image, -angle, order=1, mode="nearest")
+        debug_display(image)
+
+    if sk is not None:
+        angle = sk.skew(image)
+        note("skew", angle)
+        if angle != 0.0:
+            image = ndi.rotate(image, -angle, order=1, mode="nearest")
+        debug_display(image)
+
+    if seg is not None:
+        lines = seg.extract_textlines(image)
+        note("got", len(lines), "lines")
+
+        if rec is not None:
+            for line in lines:
+                debug_display(line["image"])
+                print rec.recognize_line(line["image"])