line-level binarization independent of region-level

pull/5/head
Konstantin Baierer 4 years ago
parent e5bc5572a2
commit a1c8f6f465

@ -79,15 +79,12 @@ class SbbBinarizeProcessor(Processor):
file_grp=self.output_file_grp) file_grp=self.output_file_grp)
page.add_AlternativeImage(AlternativeImageType(filename=bin_image_path, comment='%s,binarized' % page_xywh['features'])) page.add_AlternativeImage(AlternativeImageType(filename=bin_image_path, comment='%s,binarized' % page_xywh['features']))
else: elif oplevel == 'region':
regions = page.get_AllRegions(['Text', 'Table']) regions = page.get_AllRegions(['Text', 'Table'], depth=1)
if not regions: if not regions:
LOG.warning("Page '%s' contains no text/table regions", page_id) LOG.warning("Page '%s' contains no text/table regions", page_id)
for region in regions: for region in regions:
region_image, region_xywh = self.workspace.image_from_segment(region, page_image, page_xywh, feature_filter='binarized') region_image, region_xywh = self.workspace.image_from_segment(region, page_image, page_xywh, feature_filter='binarized')
if oplevel == 'region':
region_image_bin = self._run_binarizer(region_image) region_image_bin = self._run_binarizer(region_image)
region_image_bin_path = self.workspace.save_image_file( region_image_bin_path = self.workspace.save_image_file(
region_image_bin, region_image_bin,
@ -98,15 +95,15 @@ class SbbBinarizeProcessor(Processor):
AlternativeImageType(filename=region_image_bin_path, comments='%s,binarized' % region_xywh['features'])) AlternativeImageType(filename=region_image_bin_path, comments='%s,binarized' % region_xywh['features']))
elif oplevel == 'line': elif oplevel == 'line':
lines = region.get_TextLine() region_line_tuples = [(r.id, r.get_TextLine()) for r in page.get_AllRegions(['Text'], depth=0)]
if not lines: if not region_line_tuples:
LOG.warning("Page '%s' region '%s' contains no text lines", page_id, region.id) LOG.warning("Page '%s' contains no text lines", page_id)
for line in lines: for region_id, line in region_line_tuples:
line_image, line_xywh = self.workspace.image_from_segment(line, page_image, page_xywh, feature_filter='binarized') line_image, line_xywh = self.workspace.image_from_segment(line, page_image, page_xywh, feature_filter='binarized')
line_image_bin = self._run_binarizer(line_image) line_image_bin = self._run_binarizer(line_image)
line_image_bin_path = self.workspace.save_image_file( line_image_bin_path = self.workspace.save_image_file(
line_image_bin, line_image_bin,
"%s_%s_%s.IMG-BIN" % (file_id, region.id, line.id), "%s_%s_%s.IMG-BIN" % (file_id, region_id, line.id),
page_id=input_file.pageId, page_id=input_file.pageId,
file_grp=self.output_file_grp) file_grp=self.output_file_grp)
line.add_AlternativeImage( line.add_AlternativeImage(

Loading…
Cancel
Save