qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-26 23:34:13 +01:00
Code Issues Projects Releases Wiki Activity

order_of_regions: filter out-of-image peaks

Browse source
This commit is contained in:
Robert Sachunsky 2025-10-20 13:26:01 +02:00
parent 48761c3e12
commit c43a825d1d
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/eynollah/utils/__init__.py
View file

@ -1216,15 +1216,16 @@ def order_of_regions(textline_mask, contours_main, contours_head, y_ref):
peaks_neg, _ = find_peaks(zneg, height=0)
peaks_neg = peaks_neg - 20 - 20
##plt.plot(z)
##plt.show()
cx_main, cy_main = find_center_of_contours(contours_main)
cx_head, cy_head = find_center_of_contours(contours_head)
peaks_neg_new = np.append(np.insert(peaks_neg, 0, 0), textline_mask.shape[0])
peaks_neg_new = np.array([0] +
# peaks can be beyond box due to padding and smoothing
[peak for peak in peaks_neg
if 0 < peak and peak < textline_mask.shape[0]] +
[textline_mask.shape[0]])
# offset from bbox of mask
peaks_neg_new += y_ref
cx_main, cy_main = find_center_of_contours(contours_main)
cx_head, cy_head = find_center_of_contours(contours_head)
# assert not len(cy_main) or np.min(peaks_neg_new) <= np.min(cy_main) and np.max(cy_main) <= np.max(peaks_neg_new)
# assert not len(cy_head) or np.min(peaks_neg_new) <= np.min(cy_head) and np.max(cy_head) <= np.max(peaks_neg_new)