✨ sbb_textline_detector: Add a OCR-D interface
parent
561a6f8a90
commit
0c915c75de
@ -0,0 +1 @@
|
|||||||
|
qurator/sbb_textline_detector/ocrd-tool.json
|
@ -1 +1,2 @@
|
|||||||
from .main import *
|
from .main import *
|
||||||
|
from .ocrd_cli import *
|
||||||
|
@ -0,0 +1,19 @@
|
|||||||
|
{
|
||||||
|
"version": "0.0.1",
|
||||||
|
"tools": {
|
||||||
|
"ocrd_sbb_textline_detector": {
|
||||||
|
"executable": "ocrd_sbb_textline_detector",
|
||||||
|
"description": "Detect lines",
|
||||||
|
"steps": ["layout/segmentation/line"],
|
||||||
|
"input_file_grp": [
|
||||||
|
"OCR-D-IMG"
|
||||||
|
],
|
||||||
|
"output_file_grp": [
|
||||||
|
"OCR-D-SBB-SEG-LINE"
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"model": {"type": "string", "format": "file", "cacheable": true}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,61 @@
|
|||||||
|
import json
|
||||||
|
import os
|
||||||
|
|
||||||
|
import click
|
||||||
|
from ocrd import Processor
|
||||||
|
from ocrd.decorators import ocrd_cli_options, ocrd_cli_wrap_processor
|
||||||
|
from ocrd_utils import concat_padded, getLogger
|
||||||
|
from pkg_resources import resource_string
|
||||||
|
|
||||||
|
from qurator.sbb_textline_detector import textlineerkenner
|
||||||
|
|
||||||
|
log = getLogger('processor.OcrdSbbTextlineDetectorRecognize')
|
||||||
|
|
||||||
|
OCRD_TOOL = json.loads(resource_string(__name__, 'ocrd-tool.json').decode('utf8'))
|
||||||
|
|
||||||
|
|
||||||
|
@click.command()
|
||||||
|
@ocrd_cli_options
|
||||||
|
def ocrd_sbb_textline_detector(*args, **kwargs):
|
||||||
|
return ocrd_cli_wrap_processor(OcrdSbbTextlineDetectorRecognize, *args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
class OcrdSbbTextlineDetectorRecognize(Processor):
|
||||||
|
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
kwargs['ocrd_tool'] = OCRD_TOOL['tools']['ocrd_sbb_textline_detector']
|
||||||
|
super(OcrdSbbTextlineDetectorRecognize, self).__init__(*args, **kwargs)
|
||||||
|
|
||||||
|
def _make_file_id(self, input_file, input_file_grp, n):
|
||||||
|
file_id = input_file.ID.replace(input_file_grp, self.output_file_grp)
|
||||||
|
if file_id == input_file.ID:
|
||||||
|
file_id = concat_padded(self.output_file_grp, n)
|
||||||
|
return file_id
|
||||||
|
|
||||||
|
def process(self):
|
||||||
|
for n, page_id in enumerate(self.workspace.mets.physical_pages):
|
||||||
|
image_file = self.workspace.mets.find_files(fileGrp=self.input_file_grp, pageId=page_id)[0]
|
||||||
|
log.info("INPUT FILE %i / %s", n, image_file)
|
||||||
|
|
||||||
|
file_id = self._make_file_id(image_file, self.output_file_grp, n)
|
||||||
|
|
||||||
|
# Process the files
|
||||||
|
try:
|
||||||
|
os.mkdir(self.output_file_grp)
|
||||||
|
except FileExistsError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
model = self.parameter['model']
|
||||||
|
x = textlineerkenner(image_file.local_filename, self.output_file_grp, file_id, model)
|
||||||
|
x.run()
|
||||||
|
|
||||||
|
self.workspace.add_file(
|
||||||
|
ID=file_id + '.xml',
|
||||||
|
file_grp=self.output_file_grp,
|
||||||
|
pageId=page_id,
|
||||||
|
mimetype='application/vnd.prima.page+xml',
|
||||||
|
local_filename=self.output_file_grp + '/' + file_id)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
ocrd_sbb_textline_detector()
|
Loading…
Reference in New Issue