mirror of
https://github.com/mikegerber/ocrd_calamari.git
synced 2025-06-09 03:39:55 +02:00
recognize: remove checkpoint param in favour of checkpoint_dir alone
This commit is contained in:
parent
11615be6b2
commit
5f23c03cd9
1 changed files with 2 additions and 4 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue