From 74e23a34209f81788e4095b3e641db3d7cb25d42 Mon Sep 17 00:00:00 2001 From: Benjamin Kiessling Date: Fri, 24 Nov 2023 10:41:05 +0100 Subject: [PATCH] training regressions in transcription/segmentation --- kraken/lib/train.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/kraken/lib/train.py b/kraken/lib/train.py index a8d655e8f..29f0ef18f 100644 --- a/kraken/lib/train.py +++ b/kraken/lib/train.py @@ -357,7 +357,7 @@ def __init__(self, height, width, channels, - (self.hparams.pad, 0), + (self.hparams.hyper_params['pad'], 0), valid_norm, force_binarization) @@ -782,7 +782,7 @@ def __init__(self, height, width, channels, - self.hparams.hyper_params.padding, + self.hparams.hyper_params['padding'], valid_norm=False, force_binarization=force_binarization) @@ -809,10 +809,10 @@ def __init__(self, merge_baselines = None train_set = BaselineSet(training_data, - line_width=self.hparams.hyper_params.line_width, + line_width=self.hparams.hyper_params['line_width'], im_transforms=transforms, mode=format_type, - augmentation=self.hparams.hyper_params.augment, + augmentation=self.hparams.hyper_params['augment'], valid_baselines=valid_baselines, merge_baselines=merge_baselines, valid_regions=valid_regions, @@ -824,7 +824,7 @@ def __init__(self, if evaluation_data: val_set = BaselineSet(evaluation_data, - line_width=self.hparams.hyper_params.line_width, + line_width=self.hparams.hyper_params['line_width'], im_transforms=transforms, mode=format_type, augmentation=False,