mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-09 12:19:56 +02:00
load_model: also allow SavedModel directories
This commit is contained in:
parent
88d64de67b
commit
a69470135a
1 changed files with 2 additions and 0 deletions
|
@ -34,6 +34,8 @@ class SbbBinarizer:
|
|||
self.start_new_session()
|
||||
|
||||
self.model_files = glob('%s/*.h5' % self.model_dir)
|
||||
if not self.model_files:
|
||||
self.model_files = glob('%s/*/' % self.model_dir)
|
||||
if not self.model_files:
|
||||
raise ValueError(f"No models found in {self.model_dir}")
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue