1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Copyright 2021 Nick White.
// Use of this source code is governed by the GPLv3
// license that can be found in the LICENSE file.
// extracthocrlines copies the text and corresponding image section
// for each line of a HOCR file into separate files, which is
// useful for OCR training
package main
import (
"flag"
"fmt"
"io"
"log"
"os"
"path/filepath"
"strings"
"rescribe.xyz/utils/pkg/hocr"
"rescribe.xyz/utils/pkg/line"
)
const usage = `Usage: extracthocrlines [-b] [-d] file.hocr [file.hocr]
Copies the text and corresponding image section for each line
of a HOCR file into separate files, which is useful for OCR
training.
`
// saveline saves the text and image for a line in a directory
func saveline(l line.Detail, dir string) error {
err := os.MkdirAll(dir, 0700)
if err != nil {
return err
}
base := filepath.Join(dir, l.OcrName+"_"+l.Name)
f, err := os.Create(base + ".png")
if err != nil {
return fmt.Errorf("Error creating file %s: %v", base+".png", err)
}
defer f.Close()
err = l.Img.CopyLineTo(f)
if err != nil {
return fmt.Errorf("Error writing line image for %s: %v", base+".png", err)
}
f, err = os.Create(base + ".txt")
if err != nil {
return fmt.Errorf("Error creating file %s: %v", base+".txt", err)
}
defer f.Close()
_, err = io.WriteString(f, l.Text)
if err != nil {
return fmt.Errorf("Error writing line text for %s: %v", base+".txt", err)
}
return nil
}
func main() {
flag.Usage = func() {
fmt.Fprintf(flag.CommandLine.Output(), usage)
flag.PrintDefaults()
}
usebasepath := flag.Bool("b", false, "Use the image path of the .hocr with the .hocr suffix stripped and replaced with .png, rather than the path embedded in the .hocr")
dir := flag.String("d", ".", "Directory to save lines in")
flag.Parse()
if flag.NArg() < 1 {
flag.Usage()
os.Exit(1)
}
for _, f := range flag.Args() {
var err error
var newlines line.Details
if *usebasepath {
imgName := strings.TrimSuffix(f, ".hocr") + ".png"
newlines, err = hocr.GetLineDetailsCustomImg(f, imgName)
} else {
newlines, err = hocr.GetLineDetails(f)
}
if err != nil {
log.Fatal(err)
}
for _, l := range newlines {
if l.Img == nil {
continue
}
if l.Text == "" {
continue
}
err = saveline(l, *dir)
if err != nil {
log.Fatal(err)
}
}
}
}
|
