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

fix signature of calculate_width_height_by_columns

Browse source
This commit is contained in:
Konstantin Baierer 2021-02-01 12:55:54 +01:00
parent 0bd9a10525
commit 8f0455590a
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
sbb_newspapers_org_image/eynollah.py
View file

@ -273,7 +273,7 @@ class eynollah:
dpi = os.popen('identify -format "%x " ' + self.image_filename).read() dpi = os.popen('identify -format "%x " ' + self.image_filename).read()
return int(float(dpi)) return int(float(dpi))
def calculate_width_height_by_columns(self, img, num_col, width_early): def calculate_width_height_by_columns(self, img, num_col, width_early, label_p_pred):
if num_col == 1 and width_early < 1100: if num_col == 1 and width_early < 1100:
img_w_new = 2000 img_w_new = 2000
img_h_new = int(img.shape[0] / float(img.shape[1]) * 2000) img_h_new = int(img.shape[0] / float(img.shape[1]) * 2000)
@ -370,7 +370,7 @@ class eynollah:
gc.collect() gc.collect()
# sys.exit() # sys.exit()
img_new, num_column_is_classified = self.calculate_width_height_by_columns(img, num_col, width_early) img_new, num_column_is_classified = self.calculate_width_height_by_columns(img, num_col, width_early, label_p_pred)
if img_new.shape[1] > img.shape[1]: if img_new.shape[1] > img.shape[1]:
img_new = self.predict_enhancement(img_new) img_new = self.predict_enhancement(img_new)
@ -428,7 +428,7 @@ class eynollah:
if dpi < 298: if dpi < 298:
# sys.exit() # sys.exit()
img_new, num_column_is_classified = self.calculate_width_height_by_columns(img, num_col, width_early) img_new, num_column_is_classified = self.calculate_width_height_by_columns(img, num_col, width_early, label_p_pred)
# img_new=resize_image(img,img_h_new,img_w_new) # img_new=resize_image(img,img_h_new,img_w_new)
image_res = self.predict_enhancement(img_new) image_res = self.predict_enhancement(img_new)