summaryrefslogtreecommitdiff
path: root/pkg
diff options
context:
space:
mode:
Diffstat (limited to 'pkg')
-rw-r--r--pkg/hocr/hocr.go29
1 files changed, 29 insertions, 0 deletions
diff --git a/pkg/hocr/hocr.go b/pkg/hocr/hocr.go
index c3c88b3..fa8ae8e 100644
--- a/pkg/hocr/hocr.go
+++ b/pkg/hocr/hocr.go
@@ -127,3 +127,32 @@ func GetAvgConf(hocrfn string) (float64, error) {
}
return total / num, nil
}
+
+// GetWordConfs is a utility function that parses a hocr
+// file and returns an array containing the confidences
+// of each word therein.
+func GetWordConfs(hocrfn string) ([]float64, error) {
+ var confs []float64
+
+ file, err := ioutil.ReadFile(hocrfn)
+ if err != nil {
+ return confs, err
+ }
+
+ h, err := Parse(file)
+ if err != nil {
+ return confs, err
+ }
+
+ for _, l := range h.Lines {
+ for _, w := range l.Words {
+ c, err := wordConf(w.Title)
+ if err != nil {
+ return confs, err
+ }
+ confs = append(confs, c)
+ }
+ }
+
+ return confs, nil
+}