mirror of
https://github.com/mikegerber/ocrd_calamari.git
synced 2025-06-10 04:09:53 +02:00
use Processor.resolve_files to handle on-demand download of models via registry
This commit is contained in:
parent
d6804bd9c3
commit
00e43b1d1f
2 changed files with 3 additions and 2 deletions
|
@ -41,7 +41,8 @@ class CalamariRecognize(Processor):
|
|||
os.environ['TF_CPP_MIN_LOG_LEVEL'] = TF_CPP_MIN_LOG_LEVEL
|
||||
|
||||
if self.parameter.get('checkpoint_dir', None):
|
||||
self.parameter['checkpoint'] = '%s/*.ckpt.json' % self.parameter['checkpoint_dir']
|
||||
resolved = self.resolve_resource(self.parameter['checkpoint_dir'])
|
||||
self.parameter['checkpoint'] = '%s/*.ckpt.json' % resolved
|
||||
checkpoints = glob(self.parameter['checkpoint'])
|
||||
self.predictor = MultiPredictor(checkpoints=checkpoints)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue