qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-06-26 04:29:55 +02:00
Code Issues Releases Wiki Activity

Merge pull request #43 from apacha/fix_no_models_available

Browse source 
Raise error when no models were loaded
This commit is contained in:
vahidrezanezhad 2022-08-25 14:55:35 +02:00 • committed by GitHub
commit 2355d0ec8f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
sbb_binarize/sbb_binarize.py
View file

@ -34,7 +34,9 @@ class SbbBinarizer:
self.start_new_session()
self.model_files = glob('%s/*.h5' % self.model_dir)
if not self.model_files:
raise ValueError(f"No models found in {self.model_dir}")
self.models = []
for model_file in self.model_files:
self.models.append(self.load_model(model_file))