qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-06-09 12:19:56 +02:00
Code Issues Releases Wiki Activity

fix cv2pil ht @bertsky

Browse source
This commit is contained in:
Konstantin Baierer 2020-10-16 11:27:37 +02:00
parent eece89b9ac
commit b7a6301948
1 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
sbb_binarize/ocrd_cli.py
View file

@ -1,5 +1,3 @@
# TODO: AlternativeImage 'binarized' comment should be additive
import os.path import os.path
from pkg_resources import resource_string from pkg_resources import resource_string
from json import loads from json import loads
@ -26,8 +24,7 @@ OCRD_TOOL = loads(resource_string(__name__, 'ocrd-tool.json').decode('utf8'))
TOOL = 'ocrd-sbb-binarize' TOOL = 'ocrd-sbb-binarize'
def cv2pil(img): def cv2pil(img):
color_coverted = cv2.cvtColor(img, cv2.COLOR_BGRA2RGB) return Image.fromarray(img.as_type('uint8'))
return Image.fromarray(color_coverted)
def pil2cv(img): def pil2cv(img):
# from ocrd/workspace.py # from ocrd/workspace.py
@ -80,7 +77,7 @@ class SbbBinarizeProcessor(Processor):
file_id + '.IMG-BIN', file_id + '.IMG-BIN',
page_id=input_file.pageId, page_id=input_file.pageId,
file_grp=self.output_file_grp) file_grp=self.output_file_grp)
page.add_AlternativeImage(AlternativeImageType(filename=bin_image_path, comment=page_xywh['features']+",binarized")) page.add_AlternativeImage(AlternativeImageType(filename=bin_image_path, comment='%s,binarized' % page_xywh['features']))
else: else:
regions = page.get_AllRegions(['Text', 'Table']) regions = page.get_AllRegions(['Text', 'Table'])
@ -98,7 +95,7 @@ class SbbBinarizeProcessor(Processor):
page_id=input_file.pageId, page_id=input_file.pageId,
file_grp=self.output_file_grp) file_grp=self.output_file_grp)
region.add_AlternativeImage( region.add_AlternativeImage(
AlternativeImageType(filename=region_image_bin_path, comments=region_xywh['features']+',binarized')) AlternativeImageType(filename=region_image_bin_path, comments='%s,binarized' % region_xywh['features']))
elif oplevel == 'line': elif oplevel == 'line':
lines = region.get_TextLine() lines = region.get_TextLine()
@ -113,7 +110,7 @@ class SbbBinarizeProcessor(Processor):
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(
AlternativeImageType(filename=line_image_bin_path, comments=line_xywh['features']+',binarized')) AlternativeImageType(filename=line_image_bin_path, comments='%s,binarized' % line_xywh['features']))
self.workspace.add_file( self.workspace.add_file(
ID=file_id, ID=file_id,