From 25116a2c79440ea16d8a5e28c6f1b7f08da5c6b6 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Wed, 19 Feb 2025 00:35:48 +0100 Subject: [PATCH] resolved 2 errors --- src/eynollah/eynollah.py | 6 ++++-- src/eynollah/utils/__init__.py | 7 +++++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/src/eynollah/eynollah.py b/src/eynollah/eynollah.py index 25d5ec4..9158168 100644 --- a/src/eynollah/eynollah.py +++ b/src/eynollah/eynollah.py @@ -4074,8 +4074,10 @@ class Eynollah: ind_textline_inside_tr = list(range(len(contours[jj]))) index_textline_inside_textregion = index_textline_inside_textregion + ind_textline_inside_tr - ind_ins = [0] * len(contours[jj]) + jj - indexes_of_textline_tot = indexes_of_textline_tot + ind_ins + #ind_ins = [0] * len(contours[jj]) + jj + ind_ins = np.zeros( len(contours[jj]) ) + jj + list_ind_ins = list(ind_ins) + indexes_of_textline_tot = indexes_of_textline_tot + list_ind_ins M_main_tot = [cv2.moments(contours_txtline_of_all_textregions[j]) for j in range(len(contours_txtline_of_all_textregions))] diff --git a/src/eynollah/utils/__init__.py b/src/eynollah/utils/__init__.py index feab341..a67fc38 100644 --- a/src/eynollah/utils/__init__.py +++ b/src/eynollah/utils/__init__.py @@ -237,8 +237,11 @@ def return_x_start_end_mothers_childs_and_type_of_reading_order( if len(remained_sep_indexes)>1: #print(np.array(remained_sep_indexes),'np.array(remained_sep_indexes)') #print(np.array(mother),'mother') - remained_sep_indexes_without_mother = remained_sep_indexes[mother==0] - remained_sep_indexes_with_child_without_mother = remained_sep_indexes[mother==0 & child==1] + ##remained_sep_indexes_without_mother = remained_sep_indexes[mother==0] + ##remained_sep_indexes_with_child_without_mother = remained_sep_indexes[mother==0 & child==1] + remained_sep_indexes_without_mother=np.array(list(remained_sep_indexes))[np.array(mother)==0] + remained_sep_indexes_with_child_without_mother=np.array(list(remained_sep_indexes))[(np.array(mother)==0) & (np.array(child)==1)] + #print(remained_sep_indexes_without_mother,'remained_sep_indexes_without_mother') #print(remained_sep_indexes_without_mother,'remained_sep_indexes_without_mother') x_end_with_child_without_mother = x_end[remained_sep_indexes_with_child_without_mother]