qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-08 11:49:55 +02:00
Code Issues Projects Releases Wiki Activity

log num_cols-dependent resizing

Browse source
This commit is contained in:
Robert Sachunsky 2024-12-11 18:44:54 +00:00
parent 9270ea4550
commit b9ca7a6191
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/eynollah/eynollah.py
View file

@ -1861,6 +1861,8 @@ class Eynollah:
if num_col_classifier == 1 or num_col_classifier == 2: if num_col_classifier == 1 or num_col_classifier == 2:
if self.image_org.shape[0]/self.image_org.shape[1] > 2.5: if self.image_org.shape[0]/self.image_org.shape[1] > 2.5:
self.logger.debug("resized to %dx%d for %d cols",
img_resized.shape[1], img_resized.shape[0], num_col_classifier)
prediction_regions_org = self.do_prediction_new_concept( prediction_regions_org = self.do_prediction_new_concept(
True, img_resized, self.model_region_1_2, n_batch_inference=1, True, img_resized, self.model_region_1_2, n_batch_inference=1,
thresholding_for_some_classes_in_light_version=True) thresholding_for_some_classes_in_light_version=True)
@ -1873,6 +1875,8 @@ class Eynollah:
else: else:
new_h = (900+ (num_col_classifier-3)*100) new_h = (900+ (num_col_classifier-3)*100)
img_resized = resize_image(img_bin, int(new_h * img_bin.shape[0] /img_bin.shape[1]), new_h) img_resized = resize_image(img_bin, int(new_h * img_bin.shape[0] /img_bin.shape[1]), new_h)
self.logger.debug("resized to %dx%d (new_h=%d) for %d cols",
img_resized.shape[1], img_resized.shape[0], new_h, num_col_classifier)
prediction_regions_org = self.do_prediction_new_concept( prediction_regions_org = self.do_prediction_new_concept(
True, img_resized, self.model_region_1_2, n_batch_inference=2, True, img_resized, self.model_region_1_2, n_batch_inference=2,
thresholding_for_some_classes_in_light_version=True) thresholding_for_some_classes_in_light_version=True)