qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-04-30 19:22:03 +02:00
Code Issues Projects Releases Wiki Activity

get_marginals(): fixup point_right fallback

Browse source
This commit is contained in:
Robert Sachunsky 2026-04-26 21:52:11 +02:00
parent 68ceeec764
commit 45a43f7e5e
1 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/eynollah/utils/marginals.py
View file

@ -71,7 +71,7 @@ def get_marginals(text_mask_d, early_layout, num_col, slope_deskew, kernel=None,
# rs: also calculate the product of prominence and height (for final selection) # rs: also calculate the product of prominence and height (for final selection)
scores = np.zeros(peaks.max() + 1) scores = np.zeros(peaks.max() + 1)
scores[peaks] = props['prominences'] * props['peak_heights'] scores[peaks] = props['prominences'] * props['peak_heights']
peaks = peaks[(peaks > first_nonzero) & (peaks < last_nonzero)] peaks = peaks[(peaks > first_nonzero) & (peaks < last_nonzero)]
peaks = peaks[region_sum_0[peaks] < min_text_thickness] peaks = peaks[region_sum_0[peaks] < min_text_thickness]
@ -113,6 +113,7 @@ def get_marginals(text_mask_d, early_layout, num_col, slope_deskew, kernel=None,
else: else:
point_left = peaks_left[np.argmax(scores[peaks_left])] point_left = peaks_left[np.argmax(scores[peaks_left])]
#point_right = last_nonzero #point_right = last_nonzero
point_right = 0
main_mask_d[:, point_left: point_right] = 1 main_mask_d[:, point_left: point_right] = 1
@ -138,9 +139,7 @@ def get_marginals(text_mask_d, early_layout, num_col, slope_deskew, kernel=None,
# rs: rotate back (into undeskewed/original shape as early_layout input): # rs: rotate back (into undeskewed/original shape as early_layout input):
main_mask = rotate_image(main_mask_d, -slope_deskew) main_mask = rotate_image(main_mask_d, -slope_deskew)
main_mask_y = main_mask.sum(axis=0)
main_mask_y_nz = np.flatnonzero(main_mask_y)
min_area_text = 0.00001 min_area_text = 0.00001
main_contour = return_contours_of_interested_region(main_mask, 1, min_area_text)[0] main_contour = return_contours_of_interested_region(main_mask, 1, min_area_text)[0]
text_contours = return_contours_of_interested_region(early_layout, label_text, min_area_text) text_contours = return_contours_of_interested_region(early_layout, label_text, min_area_text)
@ -154,7 +153,7 @@ def get_marginals(text_mask_d, early_layout, num_col, slope_deskew, kernel=None,
False): False):
marg_contours.append(contour) marg_contours.append(contour)
early_layout_orig = np.copy(early_layout) # early_layout_orig = np.copy(early_layout)
early_layout = cv2.fillPoly(early_layout, pts=marg_contours, color=label_marg) early_layout = cv2.fillPoly(early_layout, pts=marg_contours, color=label_marg)
# plt.figure() # plt.figure()
@ -167,8 +166,6 @@ def get_marginals(text_mask_d, early_layout, num_col, slope_deskew, kernel=None,
# ax4 = plt.subplot(2, 2, 4, title='early_layout (undeskewed labels split)') # ax4 = plt.subplot(2, 2, 4, title='early_layout (undeskewed labels split)')
# plt.imshow(early_layout) # plt.imshow(early_layout)
# plt.show() # plt.show()
#plt.imshow(early_layout)
#plt.show()
else: else:
pass pass