diff --git a/ocrd_calamari/recognize.py b/ocrd_calamari/recognize.py index bf267d6..7620c9c 100644 --- a/ocrd_calamari/recognize.py +++ b/ocrd_calamari/recognize.py @@ -47,10 +47,8 @@ class CalamariRecognize(Processor): """ Set up the model prior to processing. """ - if not self.parameter.get('checkpoint', None) and self.parameter.get('checkpoint_dir', None): - resolved = self.resolve_resource(self.parameter['checkpoint_dir']) - self.parameter['checkpoint'] = '%s/*.ckpt.json' % resolved - checkpoints = glob(self.parameter['checkpoint']) + resolved = self.resolve_resource(self.parameter['checkpoint_dir']) + checkpoints = glob('%s/*.ckpt.json' % resolved) self.predictor = MultiPredictor(checkpoints=checkpoints) self.network_input_channels = self.predictor.predictors[0].network.input_channels