qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-07-05 17:09:54 +02:00
Code Issues Releases Wiki Activity

load_model: also allow SavedModel directories

Browse source
This commit is contained in:
Robert Sachunsky 2022-10-11 07:21:17 +00:00
parent 88d64de67b
commit a69470135a
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sbb_binarize/sbb_binarize.py
View file

@ -34,6 +34,8 @@ class SbbBinarizer:
self.start_new_session() self.start_new_session()
self.model_files = glob('%s/*.h5' % self.model_dir) 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: if not self.model_files:
raise ValueError(f"No models found in {self.model_dir}") raise ValueError(f"No models found in {self.model_dir}")