diff --git a/ocrd_calamari/recognize.py b/ocrd_calamari/recognize.py index 43b8930..9cb2238 100644 --- a/ocrd_calamari/recognize.py +++ b/ocrd_calamari/recognize.py @@ -39,9 +39,14 @@ class CalamariRecognize(Processor): kwargs['ocrd_tool'] = OCRD_TOOL['tools'][TOOL] kwargs['version'] = '%s (calamari %s, tensorflow %s)' % (OCRD_TOOL['version'], calamari_version, tensorflow_version) super(CalamariRecognize, self).__init__(*args, **kwargs) + if hasattr(self, 'output_file_grp'): + # processing context + self.setup() - def _init_calamari(self): - + def setup(self): + """ + 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 @@ -69,8 +74,6 @@ class CalamariRecognize(Processor): assert_file_grp_cardinality(self.input_file_grp, 1) assert_file_grp_cardinality(self.output_file_grp, 1) - self._init_calamari() - for (n, input_file) in enumerate(self.input_files): page_id = input_file.pageId or input_file.ID log.info("INPUT FILE %i / %s", n, page_id)