qurator-spk
/
sbb_binarization
mirror of https://github.com/qurator-spk/sbb_binarization.git
1
0
Fork 0
Code Issues Releases Wiki Activity

processor: self.resolve_resource model in addition to SBB_BINARIZE_DATA

Browse Source
pull/25/head
Konstantin Baierer 4 years ago
parent ea69494d2a
commit 4ceed759ad
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File

2
requirements.txt
Unescape Escape View File

@ -1,7 +1,7 @@
numpy >= 1.17.0, < 1.19.0 numpy >= 1.17.0, < 1.19.0
setuptools >= 41 setuptools >= 41
opencv-python-headless opencv-python-headless
ocrd >= 2.18.0 ocrd >= 2.22.3
keras >= 2.3.1, < 2.4 keras >= 2.3.1, < 2.4
h5py < 3 h5py < 3
tensorflow-gpu >= 1.15, < 1.16 tensorflow-gpu >= 1.15, < 1.16

10
sbb_binarize/ocrd_cli.py
Unescape Escape View File

@ -40,15 +40,17 @@ class SbbBinarizeProcessor(Processor):
kwargs['ocrd_tool'] = OCRD_TOOL['tools'][TOOL] kwargs['ocrd_tool'] = OCRD_TOOL['tools'][TOOL]
kwargs['version'] = OCRD_TOOL['version'] kwargs['version'] = OCRD_TOOL['version']
if not(kwargs.get('show_help', None) or kwargs.get('dump_json', None) or kwargs.get('show_version')): if not(kwargs.get('show_help', None) or kwargs.get('dump_json', None) or kwargs.get('show_version')):
LOG = getLogger('processor.SbbBinarize.__init__')
if not 'model' in kwargs['parameter']: if not 'model' in kwargs['parameter']:
raise ValueError("'model' parameter is required") raise ValueError("'model' parameter is required")
model_path = Path(kwargs['parameter']['model']) model_path = Path(kwargs['parameter']['model'])
if not model_path.is_absolute(): if not model_path.is_absolute():
if 'SBB_BINARIZE_DATA' in environ: if 'SBB_BINARIZE_DATA' in environ and environ['SBB_BINARIZE_DATA']:
LOG.info("Environment variable SBB_BINARIZE_DATA is set to '%s' - prepending to model value '%s'. If you don't want this mechanism, unset the SBB_BINARIZE_DATA environment variable.", environ['SBB_BINARIZE_DATA'], model_path)
model_path = Path(environ['SBB_BINARIZE_DATA']).joinpath(model_path) model_path = Path(environ['SBB_BINARIZE_DATA']).joinpath(model_path)
model_path = model_path.resolve() model_path = model_path.resolve()
if not model_path.is_dir(): if not model_path.is_dir():
raise FileNotFoundError("Does not exist or is not a directory: %s" % model_path) raise FileNotFoundError("Does not exist or is not a directory: %s" % model_path)
kwargs['parameter']['model'] = str(model_path) kwargs['parameter']['model'] = str(model_path)
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
@ -61,7 +63,7 @@ class SbbBinarizeProcessor(Processor):
assert_file_grp_cardinality(self.output_file_grp, 1) assert_file_grp_cardinality(self.output_file_grp, 1)
oplevel = self.parameter['operation_level'] oplevel = self.parameter['operation_level']
model_path = self.parameter['model'] # pylint: disable=attribute-defined-outside-init model_path = self.resolve_resource(self.parameter['model'])
binarizer = SbbBinarizer(model_dir=model_path, logger=LOG) binarizer = SbbBinarizer(model_dir=model_path, logger=LOG)
for n, input_file in enumerate(self.input_files): for n, input_file in enumerate(self.input_files):

Write Preview
Loading…
Cancel
Save