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(): do allow both left and right point (f/u 4bdea39)…

Browse source 
(as there are valid cases where both left and right marginalia
 is present) follow-up 4bdea39 by re-allowing left point _and_
right point - but still score-based, and not if very asymmetric
This commit is contained in:
Robert Sachunsky 2026-04-30 15:32:15 +02:00
parent 9571ce3474
commit a031d590b8
1 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -113,14 +113,17 @@ def get_marginals(text_mask, early_layout, num_col, slope_deskew,
point_left = peaks_left[np.argmax(scores[peaks_left])]
#point_right = last_nonzero
point_right = width - 1
elif scores[peaks_left].max() < scores[peaks_right].max():
point_right = peaks_right[np.argmax(scores[peaks_right])]
#point_left = first_nonzero
point_left = 0
else:
point_left = peaks_left[np.argmax(scores[peaks_left])]
#point_right = last_nonzero
point_right = 0
best_left = np.argmax(scores[peaks_left])
best_right = np.argmax(scores[peaks_right])
point_left = peaks_left[best_left]
point_right = peaks_right[best_right]
if scores[best_left] < 0.1 * scores[best_right]:
point_left = 0
#point_left = first_nonzero
if scores[best_right] < 0.1 * scores[best_left]:
point_right = 0
#point_right = last_nonzero
main_mask_d[:, point_left: point_right] = 1
if not np.any(main_mask_d):