Skip to content

Commit fe129fe

Browse files
Robert Sachunskyfinkf
authored andcommitted
segment: don't try to add if no reading order group exists
1 parent 5340d81 commit fe129fe

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

ocrd_cis/ocropy/segment.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,8 @@ def _process_element(self, element, ignore, image, coords, element_id, file_id,
523523
if line_label <= len(ignore):
524524
# existing region from `ignore` merely to be ordered
525525
# (no new region, no actual text line)
526-
index = page_add_to_reading_order(rogroup, ignore[line_label-1].id, index)
526+
if rogroup:
527+
index = page_add_to_reading_order(rogroup, ignore[line_label-1].id, index)
527528
LOG.debug('Region label %d line label %d is for ignored region "%s"',
528529
region_label, line_label, ignore[line_label-1].id)
529530
else:
@@ -545,7 +546,8 @@ def _process_element(self, element, ignore, image, coords, element_id, file_id,
545546
element.add_TextRegion(region)
546547
LOG.info('Added region "%s" with 1 line for %s "%s"',
547548
region_id, element_name, element_id)
548-
index = page_add_to_reading_order(rogroup, region.id, index)
549+
if rogroup:
550+
index = page_add_to_reading_order(rogroup, region.id, index)
549551
else:
550552
# normal case: new lines inside new regions
551553
# find contours for region (can be non-contiguous)
@@ -584,7 +586,8 @@ def _process_element(self, element, ignore, image, coords, element_id, file_id,
584586
element.add_TextRegion(region)
585587
LOG.info('Added region "%s" with %d lines for %s "%s"',
586588
region_id, len(line_polygons), element_name, element_id)
587-
index = page_add_to_reading_order(rogroup, region.id, index)
589+
if rogroup:
590+
index = page_add_to_reading_order(rogroup, region.id, index)
588591
# add additional image/non-text regions from compute_segmentation
589592
# (e.g. drop-capitals or images) ...
590593
image_labels, num_images = morph.label(images)

0 commit comments

Comments
 (0)