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

separate_lines/_vertical: gen cv2-like contours (w/ ndim=3, as in all other places)

Browse source
This commit is contained in:
Robert Sachunsky 2025-12-03 00:58:26 +01:00
parent 43a95842bd
commit ad8f8167c2
1 changed files with 64 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

128
src/eynollah/utils/separate_lines.py
View file

@ -403,14 +403,14 @@ def separate_lines(img_patch, contour_text_interest, thetha, x_help, y_help):
point_down_rot3=point_down_rot3-y_help point_down_rot3=point_down_rot3-y_help
point_down_rot4=point_down_rot4-y_help point_down_rot4=point_down_rot4-y_help
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(point_up)], textline_boxes.append(np.array([[[int(x_min), int(point_up)]],
[int(x_max), int(point_up)], [[int(x_max), int(point_up)]],
[int(x_max), int(point_down)], [[int(x_max), int(point_down)]],
[int(x_min), int(point_down)]])) [[int(x_min), int(point_down)]]]))
elif len(peaks) < 1: elif len(peaks) < 1:
pass pass
@ -462,14 +462,14 @@ def separate_lines(img_patch, contour_text_interest, thetha, x_help, y_help):
point_down_rot3=point_down_rot3-y_help point_down_rot3=point_down_rot3-y_help
point_down_rot4=point_down_rot4-y_help point_down_rot4=point_down_rot4-y_help
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(y_min)], textline_boxes.append(np.array([[[int(x_min), int(y_min)]],
[int(x_max), int(y_min)], [[int(x_max), int(y_min)]],
[int(x_max), int(y_max)], [[int(x_max), int(y_max)]],
[int(x_min), int(y_max)]])) [[int(x_min), int(y_max)]]]))
elif len(peaks) == 2: elif len(peaks) == 2:
dis_to_next = np.abs(peaks[1] - peaks[0]) dis_to_next = np.abs(peaks[1] - peaks[0])
for jj in range(len(peaks)): for jj in range(len(peaks)):
@ -530,14 +530,14 @@ def separate_lines(img_patch, contour_text_interest, thetha, x_help, y_help):
point_down_rot3=point_down_rot3-y_help point_down_rot3=point_down_rot3-y_help
point_down_rot4=point_down_rot4-y_help point_down_rot4=point_down_rot4-y_help
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(point_up)], textline_boxes.append(np.array([[[int(x_min), int(point_up)]],
[int(x_max), int(point_up)], [[int(x_max), int(point_up)]],
[int(x_max), int(point_down)], [[int(x_max), int(point_down)]],
[int(x_min), int(point_down)]])) [[int(x_min), int(point_down)]]]))
else: else:
for jj in range(len(peaks)): for jj in range(len(peaks)):
if jj == 0: if jj == 0:
@ -606,14 +606,14 @@ def separate_lines(img_patch, contour_text_interest, thetha, x_help, y_help):
point_down_rot3=point_down_rot3-y_help point_down_rot3=point_down_rot3-y_help
point_down_rot4=point_down_rot4-y_help point_down_rot4=point_down_rot4-y_help
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(point_up)], textline_boxes.append(np.array([[[int(x_min), int(point_up)]],
[int(x_max), int(point_up)], [[int(x_max), int(point_up)]],
[int(x_max), int(point_down)], [[int(x_max), int(point_down)]],
[int(x_min), int(point_down)]])) [[int(x_min), int(point_down)]]]))
return peaks, textline_boxes_rot return peaks, textline_boxes_rot
def separate_lines_vertical(img_patch, contour_text_interest, thetha): def separate_lines_vertical(img_patch, contour_text_interest, thetha):
@ -785,14 +785,14 @@ def separate_lines_vertical(img_patch, contour_text_interest, thetha):
if point_up_rot2 < 0: if point_up_rot2 < 0:
point_up_rot2 = 0 point_up_rot2 = 0
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(point_up)], textline_boxes.append(np.array([[[int(x_min), int(point_up)]],
[int(x_max), int(point_up)], [[int(x_max), int(point_up)]],
[int(x_max), int(point_down)], [[int(x_max), int(point_down)]],
[int(x_min), int(point_down)]])) [[int(x_min), int(point_down)]]]))
elif len(peaks) < 1: elif len(peaks) < 1:
pass pass
elif len(peaks) == 1: elif len(peaks) == 1:
@ -821,14 +821,14 @@ def separate_lines_vertical(img_patch, contour_text_interest, thetha):
if point_up_rot2 < 0: if point_up_rot2 < 0:
point_up_rot2 = 0 point_up_rot2 = 0
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(y_min)], textline_boxes.append(np.array([[[int(x_min), int(y_min)]],
[int(x_max), int(y_min)], [[int(x_max), int(y_min)]],
[int(x_max), int(y_max)], [[int(x_max), int(y_max)]],
[int(x_min), int(y_max)]])) [[int(x_min), int(y_max)]]]))
elif len(peaks) == 2: elif len(peaks) == 2:
dis_to_next = np.abs(peaks[1] - peaks[0]) dis_to_next = np.abs(peaks[1] - peaks[0])
for jj in range(len(peaks)): for jj in range(len(peaks)):
@ -876,14 +876,14 @@ def separate_lines_vertical(img_patch, contour_text_interest, thetha):
if point_up_rot2 < 0: if point_up_rot2 < 0:
point_up_rot2 = 0 point_up_rot2 = 0
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(point_up)], textline_boxes.append(np.array([[[int(x_min), int(point_up)]],
[int(x_max), int(point_up)], [[int(x_max), int(point_up)]],
[int(x_max), int(point_down)], [[int(x_max), int(point_down)]],
[int(x_min), int(point_down)]])) [[int(x_min), int(point_down)]]]))
else: else:
for jj in range(len(peaks)): for jj in range(len(peaks)):
if jj == 0: if jj == 0:
@ -942,14 +942,14 @@ def separate_lines_vertical(img_patch, contour_text_interest, thetha):
if point_up_rot2 < 0: if point_up_rot2 < 0:
point_up_rot2 = 0 point_up_rot2 = 0
textline_boxes_rot.append(np.array([[int(x_min_rot1), int(point_up_rot1)], textline_boxes_rot.append(np.array([[[int(x_min_rot1), int(point_up_rot1)]],
[int(x_max_rot2), int(point_up_rot2)], [[int(x_max_rot2), int(point_up_rot2)]],
[int(x_max_rot3), int(point_down_rot3)], [[int(x_max_rot3), int(point_down_rot3)]],
[int(x_min_rot4), int(point_down_rot4)]])) [[int(x_min_rot4), int(point_down_rot4)]]]))
textline_boxes.append(np.array([[int(x_min), int(point_up)], textline_boxes.append(np.array([[[int(x_min), int(point_up)]],
[int(x_max), int(point_up)], [[int(x_max), int(point_up)]],
[int(x_max), int(point_down)], [[int(x_max), int(point_down)]],
[int(x_min), int(point_down)]])) [[int(x_min), int(point_down)]]]))
return peaks, textline_boxes_rot return peaks, textline_boxes_rot
def separate_lines_new_inside_tiles2(img_patch, thetha): def separate_lines_new_inside_tiles2(img_patch, thetha):