1
0
Fork 0
mirror of https://github.com/mikegerber/ocrd_calamari.git synced 2025-06-07 19:05:01 +02:00

Merge pull request #49 from OCR-D/fix-48

check for empty line image, ht @andbue, fix #48
This commit is contained in:
Mike Gerber 2021-01-20 15:37:50 +01:00 committed by GitHub
commit a014bab5b6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -79,6 +79,7 @@ class CalamariRecognize(Processor):
line_images_np = [] line_images_np = []
for (line_no, line) in enumerate(textlines): for (line_no, line) in enumerate(textlines):
line_image, line_coords = self.workspace.image_from_segment(line, region_image, region_xywh) line_image, line_coords = self.workspace.image_from_segment(line, region_image, region_xywh)
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)
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)