qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-09 12:19:54 +02:00
Code Issues Projects Releases Wiki Activity

remove unused image_filename_stem kwarg

Browse source
This commit is contained in:
kba 2024-08-23 18:53:28 +02:00
parent 9ce02a569e
commit d6a72709a1
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
qurator/eynollah/eynollah.py
View file

@ -150,7 +150,6 @@ class Eynollah:
logger : Logger, logger : Logger,
image_filename : Optional[str] = None, image_filename : Optional[str] = None,
image_pil : Optional[Image] = None, image_pil : Optional[Image] = None,
image_filename_stem : Optional[str] = None,
dir_out : Optional[str] = None, dir_out : Optional[str] = None,
dir_in : Optional[str] = None, dir_in : Optional[str] = None,
dir_of_cropped_images : Optional[str] = None, dir_of_cropped_images : Optional[str] = None,
@ -285,6 +284,7 @@ class Eynollah:
for prefix in ('', '_grayscale'): for prefix in ('', '_grayscale'):
ret[f'img{prefix}_uint8'] = ret[f'img{prefix}'].astype(np.uint8) ret[f'img{prefix}_uint8'] = ret[f'img{prefix}'].astype(np.uint8)
return ret return ret
def reset_file_name_dir(self, image_filename): def reset_file_name_dir(self, image_filename):
self._imgs = self._cache_images(image_filename=image_filename) self._imgs = self._cache_images(image_filename=image_filename)
self.image_filename = image_filename self.image_filename = image_filename
@ -2816,12 +2816,10 @@ class Eynollah:
return polygons_of_images, img_revised_tab, text_regions_p_1_n, textline_mask_tot_d, regions_without_separators_d, regions_fully, regions_without_separators, polygons_of_marginals, contours_tables return polygons_of_images, img_revised_tab, text_regions_p_1_n, textline_mask_tot_d, regions_without_separators_d, regions_fully, regions_without_separators, polygons_of_marginals, contours_tables
def our_load_model(self, model_file): def our_load_model(self, model_file):
try: try:
model = load_model(model_file, compile=False) model = load_model(model_file, compile=False)
except: except:
model = load_model(model_file , compile=False,custom_objects = {"PatchEncoder": PatchEncoder, "Patches": Patches}) model = load_model(model_file , compile=False, custom_objects = {"PatchEncoder": PatchEncoder, "Patches": Patches})
return model return model
def run(self): def run(self):