mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-09 12:19:56 +02:00
fix cv2pil ht @bertsky
This commit is contained in:
parent
eece89b9ac
commit
b7a6301948
1 changed files with 4 additions and 7 deletions
|
@ -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,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue