qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-09 20:00:05 +02:00
Code Issues Releases Wiki Activity

update reading order machine based

Browse source
This commit is contained in:
vahidrezanezhad 2024-06-21 13:06:26 +02:00
parent 9358657a0d
commit 033cf6734b
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
generate_gt_for_training.py
View file

@ -163,8 +163,7 @@ def machine_based_reading_order(dir_xml, dir_out_modal_image, dir_out_classes, i
#print('########################') #print('########################')
xml_file = os.path.join(dir_xml,ind_xml ) xml_file = os.path.join(dir_xml,ind_xml )
f_name = ind_xml.split('.')[0] f_name = ind_xml.split('.')[0]
file_name, id_paragraph, id_header,co_text_paragraph,\ _, _, _, file_name, id_paragraph, id_header,co_text_paragraph,co_text_header,tot_region_ref,x_len, y_len,index_tot_regions,img_poly = read_xml(xml_file)
co_text_header,tot_region_ref,x_len, y_len,index_tot_regions,img_poly = read_xml(xml_file)
id_all_text = id_paragraph + id_header id_all_text = id_paragraph + id_header
co_text_all = co_text_paragraph + co_text_header co_text_all = co_text_paragraph + co_text_header