qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-03-02 13:22:00 +01:00
Code Issues Projects Releases Wiki Activity

wip

Browse source
This commit is contained in:
kba 2025-10-27 11:44:59 +01:00
parent ec1fd93dad
commit 51d2680d9c
3 changed files with 14 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/eynollah/model_zoo/default_specs.py
View file

@ -21,8 +21,8 @@ DEFAULT_MODEL_SPECS = EynollahModelSpecSet([
EynollahModelSpec(
category="binarization",
variant='',
filename="models_eynollah/eynollah-binarization-hybrid_20230504",
variant='hybrid',
filename="models_eynollah/eynollah-binarization-hybrid_20230504/model_bin_hybrid_trans_cnn_sbb_ens",
dists=['layout', 'binarization', ],
dist_url=dist_url("binarization"),
type=KerasModel,
@ -39,7 +39,7 @@ DEFAULT_MODEL_SPECS = EynollahModelSpecSet([
EynollahModelSpec(
category="binarization",
variant='augment',
variant='',
filename="models_eynollah/eynollah-binarization_20210425",
dists=['binarization'],
dist_url=dist_url("binarization"),

5
src/eynollah/sbb_binarize.py
View file

@ -27,7 +27,7 @@ class SbbBinarizer:
def __init__(self, model_dir: str, mode: str, logger=None):
if mode not in ('single', 'multi'):
raise ValueError(f"'mode' must be either 'multi' or 'single', not {mode}")
self.log = logger if logger else logging.getLogger('SbbBinarizer')
self.log = logger if logger else logging.getLogger('eynollah.binarization')
self.model_zoo = EynollahModelZoo(basedir=model_dir)
self.models = self.setup_models(mode)
self.session = self.start_new_session()
@ -51,8 +51,7 @@ class SbbBinarizer:
self.session.close()
del self.session
def predict(self, img, use_patches, n_batch_inference=5):
model = self.model_zoo.get('binarization', Model)
def predict(self, model, img, use_patches, n_batch_inference=5):
model_height = model.layers[len(model.layers)-1].output_shape[1]
model_width = model.layers[len(model.layers)-1].output_shape[2]