qurator-spk/dinglehopper
1
0
Fork 0
mirror of https://github.com/qurator-spk/dinglehopper.git synced 2025-07-11 11:29:57 +02:00
Code Issues Releases Wiki Activity

🐛 Fix generating word differences

Browse source
This commit is contained in:
Mike Gerber 2024-01-03 19:21:53 +01:00
parent 44bd4b5eda
commit c1681551af
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/dinglehopper/cli.py
View file

@ -122,9 +122,11 @@ def process(
gt_text = extract(gt, textequiv_level=textequiv_level)
ocr_text = extract(ocr, textequiv_level=textequiv_level)
gt_words = words_normalized(gt_text)
ocr_words = words_normalized(ocr_text)
gt_words: list = list(words_normalized(gt_text))
ocr_words: list = list(words_normalized(ocr_text))
assert isinstance(gt_text, ExtractedText)
assert isinstance(ocr_text, ExtractedText)
cer, n_characters = character_error_rate_n(gt_text, ocr_text)
char_diff_report, diff_c = gen_diff_report(
gt_text,
@ -136,6 +138,10 @@ def process(
differences=differences,
)
# {gt,ocr}_words must not be a generator, so we don't drain it for the differences
# report.
assert isinstance(gt_words, list)
assert isinstance(ocr_words, list)
wer, n_words = word_error_rate_n(gt_words, ocr_words)
word_diff_report, diff_w = gen_diff_report(
gt_words,