loading savedmodel

2 years ago · fa354e1e0f
parent 85f55077a4
commit fa354e1e0f
2 changed files with 5 additions and 4 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -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
--- a/sbb_binarize/sbb_binarize.py
+++ b/sbb_binarize/sbb_binarize.py
@ -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: