qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-08-09 18:19:58 +02:00
Code Issues Releases Wiki Activity

loading savedmodel

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

6
requirements.txt
View file

@ -1,5 +1,5 @@
numpy numpy
setuptools >= 41 setuptools >= 50
opencv-python-headless opencv-python-headless
ocrd >= 2.22.3 ocrd >= 2.23.3
tensorflow-gpu >= 2.6.0 tensorflow >= 2.6.0

3
sbb_binarize/sbb_binarize.py
View file

@ -91,7 +91,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)
self.model_files = glob(self.model_dir+"/*/", recursive = True)
self.models = [] self.models = []
for model_file in self.model_files: for model_file in self.model_files: