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

Merge pull request #10 from OCR-D/glob-not-listdir

Browse source 
glob *.h5 files only in model dir, fix #7
This commit is contained in:
Clemens Neudecker 2020-10-23 12:34:18 +02:00 • committed by GitHub
commit c7e8e49eac
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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 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: