mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-26 04:29:55 +02:00
loading savedmodel
This commit is contained in:
parent
85f55077a4
commit
fa354e1e0f
2 changed files with 5 additions and 4 deletions
|
@ -1,5 +1,5 @@
|
|||
numpy
|
||||
setuptools >= 41
|
||||
setuptools >= 50
|
||||
opencv-python-headless
|
||||
ocrd >= 2.22.3
|
||||
tensorflow-gpu >= 2.6.0
|
||||
ocrd >= 2.23.3
|
||||
tensorflow >= 2.6.0
|
||||
|
|
|
@ -91,7 +91,8 @@ class SbbBinarizer:
|
|||
|
||||
self.start_new_session()
|
||||
|
||||
self.model_files = glob('%s/*.h5' % self.model_dir)
|
||||
#self.model_files = glob('%s/*.h5' % self.model_dir)
|
||||
self.model_files = glob(self.model_dir+"/*/", recursive = True)
|
||||
|
||||
self.models = []
|
||||
for model_file in self.model_files:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue