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

Merge pull request #6 from OCR-D/envvar

Browse source 
ocrd cli: "model" parameter replaceable by SBB_BINARIZE_DATA envvar
This commit is contained in:
vahidrezanezhad 2020-10-23 00:15:41 +02:00 • committed by GitHub
commit d675ffc066
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sbb_binarize/ocrd-tool.json
View file

@ -19,7 +19,7 @@
"model": { "model": {
"description": "models directory.", "description": "models directory.",
"type": "string", "type": "string",
"required": true "required": false
} }
} }
} }

8
sbb_binarize/ocrd_cli.py
View file

@ -37,6 +37,14 @@ class SbbBinarizeProcessor(Processor):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
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 'parameter' in kwargs:
kwargs['parameter'] = {}
if not 'model' in kwargs['parameter']:
if 'SBB_BINARIZE_DATA' in os.environ:
kwargs['parameter']['model'] = os.environ['SBB_BINARIZE_DATA']
else:
raise ValueError("Must pass 'model' parameter or set SBB_BINARIZE_DATA environment variable")
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def process(self): def process(self):