qurator-spk/sbb_textline_detection
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_textline_detection.git synced 2025-10-31 09:24:17 +01:00
Code Issues Releases Wiki Activity

resolve 2020-02-17-bug-sbb_textline_detector

Browse source
This commit is contained in:
b-vr103 2020-02-21 16:48:27 +01:00
parent 689822c63a
commit b9caa8e12c
1 changed files with 13 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

29
qurator/sbb_textline_detector/main.py
View file

@ -1714,8 +1714,7 @@ class textline_detector:
return ang_int
def do_work_of_slopes(self,queue_of_slopes_per_textregion,queue_of_textlines_rectangle_per_textregion
,queue_of_textregion_box,boxes_per_process,queue_of_quntours_of_textregion,textline_mask_tot,contours_per_process):
def do_work_of_slopes(self,queue_of_all_params,boxes_per_process,textline_mask_tot,contours_per_process):
slopes_per_each_subprocess = []
bounding_box_of_textregion_per_each_subprocess=[]
@ -1752,19 +1751,11 @@ class textline_detector:
bounding_box_of_textregion_per_each_subprocess.append(boxes_per_process[mv] )
queue_of_slopes_per_textregion.put(slopes_per_each_subprocess)
queue_of_textlines_rectangle_per_textregion.put(textlines_rectangles_per_each_subprocess)
queue_of_textregion_box.put(bounding_box_of_textregion_per_each_subprocess )
queue_of_quntours_of_textregion.put(contours_textregion_per_each_subprocess)
queue_of_all_params.put([slopes_per_each_subprocess, textlines_rectangles_per_each_subprocess, bounding_box_of_textregion_per_each_subprocess, contours_textregion_per_each_subprocess])
def get_slopes_and_deskew(self, contours,textline_mask_tot):
num_cores =cpu_count()
queue_of_slopes_per_textregion = Queue()
queue_of_textlines_rectangle_per_textregion=Queue()
queue_of_textregion_box=Queue()
queue_of_quntours_of_textregion=Queue()
queue_of_all_params = Queue()
processes = []
nh=np.linspace(0, len(self.boxes), num_cores+1)
@ -1773,8 +1764,8 @@ class textline_detector:
for i in range(num_cores):
boxes_per_process=self.boxes[int(nh[i]):int(nh[i+1])]
contours_per_process=contours[int(nh[i]):int(nh[i+1])]
processes.append(Process(target=self.do_work_of_slopes, args=(queue_of_slopes_per_textregion,queue_of_textlines_rectangle_per_textregion,
queue_of_textregion_box, boxes_per_process, queue_of_quntours_of_textregion, textline_mask_tot, contours_per_process)))
processes.append(Process(target=self.do_work_of_slopes, args=(queue_of_all_params, boxes_per_process, textline_mask_tot, contours_per_process)))
for i in range(num_cores):
processes[i].start()
@ -1785,10 +1776,12 @@ class textline_detector:
self.boxes=[]
for i in range(num_cores):
slopes_for_sub_process=queue_of_slopes_per_textregion.get(True)
boxes_for_sub_process=queue_of_textregion_box.get(True)
polys_for_sub_process=queue_of_textlines_rectangle_per_textregion.get(True)
contours_for_subprocess=queue_of_quntours_of_textregion.get(True)
list_all_par=queue_of_all_params.get(True)
slopes_for_sub_process=list_all_par[0]
polys_for_sub_process=list_all_par[1]
boxes_for_sub_process=list_all_par[2]
contours_for_subprocess=list_all_par[3]
for j in range(len(slopes_for_sub_process)):
self.slopes.append(slopes_for_sub_process[j])