compatibility forced_align_overlay.py XML output with container classes

mittagessen · Apr 24, 2024 · 786fcb9 · 786fcb9
1 parent 3a8aba1
commit 786fcb9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kraken/contrib/forced_alignment_overlay.py b/kraken/contrib/forced_alignment_overlay.py
@@ -50,7 +50,9 @@ def _repl_alto(fname, cuts):
                     for chld in el:
                         if chld.tag.endswith('Glyph'):
                             el.remove(chld)
-                    for char in line_cuts[idx:str_len]:
+                    for char in zip(line_cuts.prediction[idx:str_len],
+                                    line_cuts.cuts[idx:str_len],
+                                    line_cuts.confidences[idx:str_len]):
                         glyph = etree.SubElement(el, 'Glyph')
                         glyph.set('ID', f'char_{char_idx}')
                         char_idx += 1