fix divide zero error

4 years ago · dc0a21ffc7
parent 23df5bcf12
commit dc0a21ffc7
1 changed files with 1 additions and 1 deletions
--- a/ppocr/metrics/rec_metric.py
+++ b/ppocr/metrics/rec_metric.py
@ -29,7 +29,7 @@ class RecMetric(object):
            pred = pred.replace(" ", "")
            target = target.replace(" ", "")
            norm_edit_dis += Levenshtein.distance(pred, target) / max(
-                len(pred), len(target))
+                len(pred), len(target), 1)
            if pred == target:
                correct_num += 1
            all_num += 1