summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--bookpipeline/cmd/bookpipeline/main.go6
-rw-r--r--lib/hocr/hocr.go2
2 files changed, 4 insertions, 4 deletions
diff --git a/bookpipeline/cmd/bookpipeline/main.go b/bookpipeline/cmd/bookpipeline/main.go
index 97f4109..ec6a08d 100644
--- a/bookpipeline/cmd/bookpipeline/main.go
+++ b/bookpipeline/cmd/bookpipeline/main.go
@@ -149,6 +149,9 @@ func analyse(toanalyse chan string, up chan string, errc chan error, logger *log
}
logger.Println("Calculating confidence for", path)
avg, err := hocr.GetAvgConf(path)
+ if err != nil && err.Error() == "No words found" {
+ continue
+ }
if err != nil {
for range toanalyse {
} // consume the rest of the receiving channel so it isn't blocked
@@ -156,9 +159,6 @@ func analyse(toanalyse chan string, up chan string, errc chan error, logger *log
errc <- errors.New(fmt.Sprintf("Error retreiving confidence for %s: %s", path, err))
return
}
- if avg == 0 {
- continue
- }
base := filepath.Base(path)
codestart := strings.Index(base, "_bin")
name := base[0:codestart]
diff --git a/lib/hocr/hocr.go b/lib/hocr/hocr.go
index 052aa19..dcd0494 100644
--- a/lib/hocr/hocr.go
+++ b/lib/hocr/hocr.go
@@ -123,7 +123,7 @@ func GetAvgConf(hocrfn string) (float64, error) {
}
}
if num == 0 {
- return 0, nil
+ return 0, errors.New("No words found")
}
return total / num, nil
}