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

pull/10/head
Konstantin Baierer 4 years ago
parent d675ffc066
commit 664a441a2d

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

Loading…
Cancel
Save