mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-30 22:49:54 +02:00
add OCR-D click interface
This commit is contained in:
parent
6913415f55
commit
d4c97ebf72
2 changed files with 9 additions and 1 deletions
|
@ -4,15 +4,17 @@ import os.path
|
||||||
from pkg_resources import resource_string
|
from pkg_resources import resource_string
|
||||||
from json import loads
|
from json import loads
|
||||||
|
|
||||||
|
from click import command
|
||||||
from ocrd_utils import (
|
from ocrd_utils import (
|
||||||
getLogger,
|
getLogger,
|
||||||
assert_file_grp_cardinality,
|
assert_file_grp_cardinality,
|
||||||
make_file_id,
|
make_file_id,
|
||||||
MIMETYPE_PAGE
|
MIMETYPE_PAGE
|
||||||
)
|
)
|
||||||
|
from ocrd import Processor
|
||||||
from ocrd_modelfactory import page_from_file
|
from ocrd_modelfactory import page_from_file
|
||||||
from ocrd_models.ocrd_page import AlternativeImageType, to_xml
|
from ocrd_models.ocrd_page import AlternativeImageType, to_xml
|
||||||
from ocrd import Processor
|
from ocrd.decorators import ocrd_cli_options, ocrd_cli_wrap_processor
|
||||||
|
|
||||||
from .sbb_binarize import SbbBinarizer
|
from .sbb_binarize import SbbBinarizer
|
||||||
|
|
||||||
|
@ -113,3 +115,8 @@ class SbbBinarizeProcessor(Processor):
|
||||||
mimetype=MIMETYPE_PAGE,
|
mimetype=MIMETYPE_PAGE,
|
||||||
local_filename=os.path.join(self.output_file_grp, file_id + '.xml'),
|
local_filename=os.path.join(self.output_file_grp, file_id + '.xml'),
|
||||||
content=to_xml(pcgts))
|
content=to_xml(pcgts))
|
||||||
|
|
||||||
|
@command()
|
||||||
|
@ocrd_cli_options
|
||||||
|
def cli(*args, **kwargs):
|
||||||
|
return ocrd_cli_wrap_processor(SbbBinarizeProcessor, *args, **kwargs)
|
||||||
|
|
1
setup.py
1
setup.py
|
@ -23,6 +23,7 @@ setup(
|
||||||
entry_points={
|
entry_points={
|
||||||
'console_scripts': [
|
'console_scripts': [
|
||||||
'sbb_binarize=sbb_binarize.cli:main',
|
'sbb_binarize=sbb_binarize.cli:main',
|
||||||
|
'ocrd-sbb-binarize=sbb_binarize.ocrd_cli:cli',
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue