Skip to content

Commit

Permalink
issue #67 solved
Browse files Browse the repository at this point in the history
  • Loading branch information
vahid committed May 19, 2023
1 parent a96147b commit 45c40a5
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion qurator/eynollah/eynollah.py
Original file line number Diff line number Diff line change
Expand Up @@ -452,6 +452,9 @@ def calculate_width_height_by_columns(self, img, num_col, width_early, label_p_p
if label_p_pred[0][int(num_col - 1)] < 0.9 and img_w_new < width_early:
img_new = np.copy(img)
num_column_is_classified = False
elif label_p_pred[0][int(num_col - 1)] < 0.8 and img_h_new >= 8000:
img_new = np.copy(img)
num_column_is_classified = False
else:
img_new = resize_image(img, img_h_new, img_w_new)
num_column_is_classified = True
Expand Down Expand Up @@ -2831,7 +2834,7 @@ def run(self):
self.reset_file_name_dir(os.path.join(self.dir_in,img_name))

img_res, is_image_enhanced, num_col_classifier, num_column_is_classified = self.run_enhancement(self.light_version)

print(img_res.shape)
self.logger.info("Enhancing took %.1fs ", time.time() - t0)

t1 = time.time()
Expand Down

0 comments on commit 45c40a5

Please sign in to comment.