Skip to content

Commit 2c0c4be

Browse files
repair bleu score computation (#14626)
1 parent ebd08eb commit 2c0c4be

File tree

5 files changed

+9
-6
lines changed

5 files changed

+9
-6
lines changed

configs/rec/PP-FormuaNet/rec_pp_formulanet_l.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ PostProcess:
6969
Metric:
7070
name: LaTeXOCRMetric
7171
main_indicator: exp_rate
72-
cal_bleu_score: False
72+
cal_bleu_score: True
7373

7474
Train:
7575
dataset:

configs/rec/PP-FormuaNet/rec_pp_formulanet_s.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ Architecture:
5353
encoder_hidden_size: 2048
5454
is_export: False
5555
length_aware: True
56-
use_parallel: True,
56+
use_parallel: True
5757
parallel_step: 3
5858

5959
Loss:
@@ -67,7 +67,7 @@ PostProcess:
6767
Metric:
6868
name: LaTeXOCRMetric
6969
main_indicator: exp_rate
70-
cal_bleu_score: False
70+
cal_bleu_score: True
7171

7272
Train:
7373
dataset:

configs/rec/rec_latex_ocr.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ PostProcess:
6666
Metric:
6767
name: LaTeXOCRMetric
6868
main_indicator: exp_rate
69-
cal_bleu_score: False
69+
cal_bleu_score: True
7070

7171
Train:
7272
dataset:

configs/rec/rec_unimernet.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ PostProcess:
6666
Metric:
6767
name: LaTeXOCRMetric
6868
main_indicator: exp_rate
69-
cal_bleu_score: False
69+
cal_bleu_score: True
7070

7171
Train:
7272
dataset:

ppocr/metrics/bleu.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,10 @@ def compute_bleu(reference_corpus, translation_corpus, max_order=4, smooth=False
9898
else:
9999
geo_mean = 0
100100

101-
ratio = float(translation_length) / reference_length
101+
if float(translation_length) == 0 or float(reference_length) == 0:
102+
ratio = 1e-5
103+
else:
104+
ratio = float(translation_length) / reference_length
102105

103106
if ratio > 1.0:
104107
bp = 1.0

0 commit comments

Comments
 (0)