loading savedmodel

This commit is contained in:
vahid 2023-05-04 11:08:56 +02:00
parent 85f55077a4
commit fa354e1e0f
2 changed files with 5 additions and 4 deletions

View file

@ -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

View file

@ -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: