qurator-spk/ocrd_calamari
1
0
Fork 0
mirror of https://github.com/mikegerber/ocrd_calamari.git synced 2025-11-04 09:34:13 +01:00
Code Releases Activity

🐛 Fix word and glyph coordinates

Browse source 
Fixes GH-57.
This commit is contained in:
Gerber, Mike 2021-02-09 19:57:03 +01:00
parent 41a48455f8
commit 0869386ec4
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ocrd_calamari/recognize.py
View file

@ -87,6 +87,7 @@ class CalamariRecognize(Processor):
textlines = region.get_TextLine() textlines = region.get_TextLine()
log.info("About to recognize %i lines of region '%s'", len(textlines), region.id) log.info("About to recognize %i lines of region '%s'", len(textlines), region.id)
line_images_np = [] line_images_np = []
line_coordss = []
for line in textlines: for line in textlines:
log.debug("Recognizing line '%s' in region '%s'", line.id, region.id) log.debug("Recognizing line '%s' in region '%s'", line.id, region.id)
@ -101,9 +102,10 @@ class CalamariRecognize(Processor):
line_image = line_image if all(line_image.size) else [[0]] line_image = line_image if all(line_image.size) else [[0]]
line_image_np = np.array(line_image, dtype=np.uint8) line_image_np = np.array(line_image, dtype=np.uint8)
line_images_np.append(line_image_np) line_images_np.append(line_image_np)
line_coordss.append(line_coords)
raw_results_all = self.predictor.predict_raw(line_images_np, progress_bar=False) raw_results_all = self.predictor.predict_raw(line_images_np, progress_bar=False)
for line, raw_results in zip(textlines, raw_results_all): for line, line_coords, raw_results in zip(textlines, line_coordss, raw_results_all):
for i, p in enumerate(raw_results): for i, p in enumerate(raw_results):
p.prediction.id = "fold_{}".format(i) p.prediction.id = "fold_{}".format(i)