|
|
@ -1,53 +1,43 @@
|
|
|
|
import json
|
|
|
|
from functools import cached_property
|
|
|
|
import os
|
|
|
|
import os
|
|
|
|
|
|
|
|
from typing import Optional
|
|
|
|
|
|
|
|
|
|
|
|
import click
|
|
|
|
import click
|
|
|
|
import importlib_resources
|
|
|
|
from ocrd_models import OcrdFileType
|
|
|
|
from ocrd import Processor
|
|
|
|
from ocrd import Processor
|
|
|
|
from ocrd.decorators import ocrd_cli_options, ocrd_cli_wrap_processor
|
|
|
|
from ocrd.decorators import ocrd_cli_options, ocrd_cli_wrap_processor
|
|
|
|
from ocrd_utils import assert_file_grp_cardinality, getLogger, make_file_id
|
|
|
|
from ocrd_utils import make_file_id
|
|
|
|
|
|
|
|
|
|
|
|
from .cli import process as cli_process
|
|
|
|
from .cli import process as cli_process
|
|
|
|
|
|
|
|
|
|
|
|
OCRD_TOOL = json.loads(
|
|
|
|
|
|
|
|
importlib_resources.files(__name__)
|
|
|
|
|
|
|
|
.joinpath("ocrd-tool.json")
|
|
|
|
|
|
|
|
.read_text(encoding="utf-8", errors="strict")
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@click.command()
|
|
|
|
@click.command()
|
|
|
|
@ocrd_cli_options
|
|
|
|
@ocrd_cli_options
|
|
|
|
def ocrd_dinglehopper(*args, **kwargs):
|
|
|
|
def ocrd_dinglehopper(*args, **kwargs):
|
|
|
|
return ocrd_cli_wrap_processor(OcrdDinglehopperEvaluate, *args, **kwargs)
|
|
|
|
return ocrd_cli_wrap_processor(OcrdDinglehopperEvaluate, *args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class OcrdDinglehopperEvaluate(Processor):
|
|
|
|
class OcrdDinglehopperEvaluate(Processor):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
|
|
|
|
kwargs["ocrd_tool"] = OCRD_TOOL["tools"]["ocrd-dinglehopper"]
|
|
|
|
|
|
|
|
kwargs["version"] = OCRD_TOOL["version"]
|
|
|
|
|
|
|
|
super(OcrdDinglehopperEvaluate, self).__init__(*args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def process(self):
|
|
|
|
@cached_property
|
|
|
|
assert_file_grp_cardinality(self.input_file_grp, 2, "GT and OCR")
|
|
|
|
def executable(self):
|
|
|
|
assert_file_grp_cardinality(self.output_file_grp, 1)
|
|
|
|
return 'ocrd-dinglehopper'
|
|
|
|
|
|
|
|
|
|
|
|
log = getLogger("processor.OcrdDinglehopperEvaluate")
|
|
|
|
def process_page_file(self, *input_files: Optional[OcrdFileType]) -> None:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
assert self.parameter
|
|
|
|
metrics = self.parameter["metrics"]
|
|
|
|
metrics = self.parameter["metrics"]
|
|
|
|
textequiv_level = self.parameter["textequiv_level"]
|
|
|
|
textequiv_level = self.parameter["textequiv_level"]
|
|
|
|
gt_grp, ocr_grp = self.input_file_grp.split(",")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
input_file_tuples = self.zip_input_files(on_error="abort")
|
|
|
|
try:
|
|
|
|
for n, (gt_file, ocr_file) in enumerate(input_file_tuples):
|
|
|
|
gt_file, ocr_file = input_files
|
|
|
|
if not gt_file or not ocr_file:
|
|
|
|
assert gt_file, 'missing GT file'
|
|
|
|
# file/page was not found in this group
|
|
|
|
assert ocr_file, 'missing OCR file'
|
|
|
|
continue
|
|
|
|
assert gt_file.local_filename
|
|
|
|
gt_file = self.workspace.download_file(gt_file)
|
|
|
|
assert ocr_file.local_filename
|
|
|
|
ocr_file = self.workspace.download_file(ocr_file)
|
|
|
|
except (ValueError, AssertionError) as err:
|
|
|
|
page_id = gt_file.pageId
|
|
|
|
self.logger.warning(f'Missing either GT file, OCR file or both: {err}') # TODO how to log which page?
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
log.info("INPUT FILES %i / %s↔ %s", n, gt_file, ocr_file)
|
|
|
|
page_id = gt_file.pageId
|
|
|
|
|
|
|
|
|
|
|
|
file_id = make_file_id(ocr_file, self.output_file_grp)
|
|
|
|
file_id = make_file_id(ocr_file, self.output_file_grp)
|
|
|
|
report_prefix = os.path.join(self.output_file_grp, file_id)
|
|
|
|
report_prefix = os.path.join(self.output_file_grp, file_id)
|
|
|
|