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

glob *.h5 files only in model dir, fix #7

Browse source
This commit is contained in:
Konstantin Baierer 2020-10-23 11:30:31 +02:00
parent d675ffc066
commit 664a441a2d
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
sbb_binarize/sbb_binarize.py
View file

@ -3,7 +3,8 @@ Tool to load model and binarize a given image.
"""
import sys
from os import listdir, environ, devnull
from glob import glob
from os import environ, devnull
from os.path import join
from warnings import catch_warnings, simplefilter
@ -191,7 +192,7 @@ class SbbBinarizer:
if image_path is not None:
image = cv2.imread(image)
self.start_new_session()
list_of_model_files = listdir(self.model_dir)
list_of_model_files = glob('%s/*.h5' % self.model_dir)
img_last = 0
for model_in in list_of_model_files: