qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-09 20:29:55 +02:00
Code Issues Projects Releases Wiki Activity

class Eynollah: add typing, consistent interface in CLI and OCR-D CLI

Browse source
This commit is contained in:
kba 2024-08-23 18:32:29 +02:00
parent 0a3f525f0a
commit 4a13781ef4
2 changed files with 34 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

5
qurator/eynollah/cli.py
View file

@ -1,6 +1,6 @@
import sys import sys
import click import click
from ocrd_utils import initLogging, setOverrideLogLevel from ocrd_utils import getLogger, initLogging, setOverrideLogLevel
from qurator.eynollah.eynollah import Eynollah from qurator.eynollah.eynollah import Eynollah
@ -176,10 +176,11 @@ def main(
print('Error: You used -tll to enable light textline detection but -light is not enabled') print('Error: You used -tll to enable light textline detection but -light is not enabled')
sys.exit(1) sys.exit(1)
eynollah = Eynollah( eynollah = Eynollah(
model,
getLogger('Eynollah'),
image_filename=image, image_filename=image,
dir_out=out, dir_out=out,
dir_in=dir_in, dir_in=dir_in,
dir_models=model,
dir_of_cropped_images=save_images, dir_of_cropped_images=save_images,
dir_of_layout=save_layout, dir_of_layout=save_layout,
dir_of_deskewed=save_deskewed, dir_of_deskewed=save_deskewed,

58
qurator/eynollah/eynollah.py
View file

@ -6,14 +6,18 @@
document layout analysis (segmentation) with output in PAGE-XML document layout analysis (segmentation) with output in PAGE-XML
""" """
from logging import Logger
import math import math
import os import os
import sys import sys
import time import time
from typing import Optional
import warnings import warnings
from pathlib import Path from pathlib import Path
from multiprocessing import Process, Queue, cpu_count from multiprocessing import Process, Queue, cpu_count
import gc import gc
from PIL.Image import Image
from ocrd import OcrdPage
from ocrd_utils import getLogger from ocrd_utils import getLogger
import cv2 import cv2
import numpy as np import numpy as np
@ -142,32 +146,32 @@ class PatchEncoder(layers.Layer):
class Eynollah: class Eynollah:
def __init__( def __init__(
self, self,
dir_models, dir_models : str,
image_filename=None, logger : Logger,
image_pil=None, image_filename : Optional[str] = None,
image_filename_stem=None, image_pil : Optional[Image] = None,
dir_out=None, image_filename_stem : Optional[str] = None,
dir_in=None, dir_out : Optional[str] = None,
dir_of_cropped_images=None, dir_in : Optional[str] = None,
dir_of_layout=None, dir_of_cropped_images : Optional[str] = None,
dir_of_deskewed=None, dir_of_layout : Optional[str] = None,
dir_of_all=None, dir_of_deskewed : Optional[str] = None,
dir_save_page=None, dir_of_all : Optional[str] = None,
enable_plotting=False, dir_save_page : Optional[str] = None,
allow_enhancement=False, enable_plotting : bool = False,
curved_line=False, allow_enhancement : bool = False,
textline_light=False, curved_line : bool = False,
full_layout=False, textline_light : bool = False,
tables=False, full_layout : bool = False,
right2left=False, tables : bool = False,
input_binary=False, right2left : bool = False,
allow_scaling=False, input_binary : bool = False,
headers_off=False, allow_scaling : bool = False,
light_version=False, headers_off : bool = False,
ignore_page_extraction=False, light_version : bool = False,
override_dpi=None, ignore_page_extraction : bool = False,
logger=None, override_dpi : Optional[int] = None,
pcgts=None, pcgts : Optional[OcrdPage] = None,
): ):
if not dir_in: if not dir_in:
if image_pil: if image_pil:
@ -213,7 +217,7 @@ class Eynollah:
curved_line=self.curved_line, curved_line=self.curved_line,
textline_light = self.textline_light, textline_light = self.textline_light,
pcgts=pcgts) pcgts=pcgts)
self.logger = logger if logger else getLogger('eynollah') self.logger = logger
self.dir_models = dir_models self.dir_models = dir_models
self.model_dir_of_enhancement = dir_models + "/eynollah-enhancement_20210425" self.model_dir_of_enhancement = dir_models + "/eynollah-enhancement_20210425"