diff options
Diffstat (limited to 'pkg')
-rw-r--r-- | pkg/hocr/hocr.go | 9 | ||||
-rw-r--r-- | pkg/hocr/lines.go | 5 | ||||
-rw-r--r-- | pkg/line/line.go | 1 | ||||
-rw-r--r-- | pkg/prob/prob.go | 3 |
4 files changed, 14 insertions, 4 deletions
diff --git a/pkg/hocr/hocr.go b/pkg/hocr/hocr.go index 9dea49c..6b43558 100644 --- a/pkg/hocr/hocr.go +++ b/pkg/hocr/hocr.go @@ -2,6 +2,8 @@ // Use of this source code is governed by the GPLv3 // license that can be found in the LICENSE file. +// hocr contains structures and functions for parsing and analysing +// hocr files package hocr import ( @@ -51,6 +53,7 @@ func wordConf(s string) (float64, error) { return strconv.ParseFloat(conf[1], 64) } +// BoxCoords parses bbox coordinate strings func BoxCoords(s string) ([4]int, error) { var coords [4]int re, err := regexp.Compile(`bbox ([0-9]+) ([0-9]+) ([0-9]+) ([0-9]+)`) @@ -73,6 +76,7 @@ func noText(s string) bool { return len(t) == 0 } +// Parse parses a hOCR file func Parse(b []byte) (Hocr, error) { var hocr Hocr @@ -84,6 +88,7 @@ func Parse(b []byte) (Hocr, error) { return hocr, nil } +// GetText parses a hOCR file and extracts the text from it func GetText(hocrfn string) (string, error) { var s string @@ -104,6 +109,8 @@ func GetText(hocrfn string) (string, error) { return s, nil } +// GetAvgConf calculates the average confidence of a hOCR file from +// confidences embedded in each word func GetAvgConf(hocrfn string) (float64, error) { file, err := ioutil.ReadFile(hocrfn) if err != nil { @@ -134,7 +141,7 @@ func GetAvgConf(hocrfn string) (float64, error) { // GetWordConfs is a utility function that parses a hocr // file and returns an array containing the confidences -// of each word therein. +// of each word therein func GetWordConfs(hocrfn string) ([]float64, error) { var confs []float64 diff --git a/pkg/hocr/lines.go b/pkg/hocr/lines.go index 1387574..942bd01 100644 --- a/pkg/hocr/lines.go +++ b/pkg/hocr/lines.go @@ -20,6 +20,7 @@ import ( "rescribe.xyz/utils/pkg/line" ) +// LineText extracts the text from an OcrLine func LineText(l OcrLine) (string) { linetext := "" @@ -88,7 +89,7 @@ func parseLineDetails(h Hocr, i *image.Gray, name string) (line.Details, error) } // GetLineDetails parses a hocr file and returns a corresponding -// line.Details, including image extracts for each line. +// line.Details, including image extracts for each line func GetLineDetails(hocrfn string) (line.Details, error) { var newlines line.Details @@ -121,7 +122,7 @@ func GetLineDetails(hocrfn string) (line.Details, error) { } // GetLineBasics parses a hocr file and returns a corresponding -// line.Details, without any image extracts. +// line.Details, without any image extracts func GetLineBasics(hocrfn string) (line.Details, error) { var newlines line.Details diff --git a/pkg/line/line.go b/pkg/line/line.go index b1e9021..bac7fa4 100644 --- a/pkg/line/line.go +++ b/pkg/line/line.go @@ -2,6 +2,7 @@ // Use of this source code is governed by the GPLv3 // license that can be found in the LICENSE file. +// line contains various functions to manipulate ocr lines package line import ( diff --git a/pkg/prob/prob.go b/pkg/prob/prob.go index 7a6979a..67baa03 100644 --- a/pkg/prob/prob.go +++ b/pkg/prob/prob.go @@ -2,6 +2,7 @@ // Use of this source code is governed by the GPLv3 // license that can be found in the LICENSE file. +// prob processes .prob files generated by ocropus package prob import ( @@ -41,7 +42,7 @@ func getLineAvg(f string) (float64, error) { return avg, nil } -// Note this only processes one line at a time +// GetLineDetails parses a .prob and corresponding .txt file func GetLineDetails(probfn string) (line.Details, error) { var l line.Detail lines := make(line.Details, 0) |