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

loading savedmodel format and saving as png

Browse source
This commit is contained in:
vahidrezanezhad 2024-08-20 15:46:20 +02:00
parent 128aa8436d
commit 3095498162
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

@ -34,7 +34,7 @@ class SbbBinarizer:
self.start_new_session() self.start_new_session()
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:
@ -278,6 +278,7 @@ class SbbBinarizer:
else: else:
ls_imgs = os.listdir(dir_in) ls_imgs = os.listdir(dir_in)
for image_name in ls_imgs: for image_name in ls_imgs:
image_stem = image_name.split('.')[0]
print(image_name,'image_name') print(image_name,'image_name')
image = cv2.imread(os.path.join(dir_in,image_name) ) image = cv2.imread(os.path.join(dir_in,image_name) )
img_last = 0 img_last = 0
@ -302,4 +303,4 @@ class SbbBinarizer:
img_last[:, :][img_last[:, :] > 0] = 255 img_last[:, :][img_last[:, :] > 0] = 255
img_last = (img_last[:, :] == 0) * 255 img_last = (img_last[:, :] == 0) * 255
cv2.imwrite(os.path.join(dir_out,image_name), img_last) cv2.imwrite(os.path.join(dir_out,image_stem+'.png'), img_last)