just defined graphic region types can be extracted as label

pull/18/head
vahidrezanezhad 7 months ago
parent b1d971a200
commit dc356a5f42

@ -327,6 +327,7 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
co_text = {'drop-capital':[], "footnote":[], "footnote-continued":[], "heading":[], "signature-mark":[], "header":[], "catch-word":[], "page-number":[], "marginalia":[], "paragraph":[]} co_text = {'drop-capital':[], "footnote":[], "footnote-continued":[], "heading":[], "signature-mark":[], "header":[], "catch-word":[], "page-number":[], "marginalia":[], "paragraph":[]}
all_defined_textregion_types = list(co_text.keys()) all_defined_textregion_types = list(co_text.keys())
co_graphic = {"handwritten-annotation":[], "decoration":[], "stamp":[], "signature":[]} co_graphic = {"handwritten-annotation":[], "decoration":[], "stamp":[], "signature":[]}
all_defined_graphic_types = list(co_graphic.keys())
co_sep=[] co_sep=[]
co_img=[] co_img=[]
co_table=[] co_table=[]
@ -425,6 +426,7 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
else: else:
if "type" in nn.attrib: if "type" in nn.attrib:
if nn.attrib['type'] in all_defined_graphic_types:
c_t_in_graphic[nn.attrib['type']].append( np.array( [ [ int(x.split(',')[0]) , int(x.split(',')[1]) ] for x in p_h] ) ) c_t_in_graphic[nn.attrib['type']].append( np.array( [ [ int(x.split(',')[0]) , int(x.split(',')[1]) ] for x in p_h] ) )
break break
@ -450,6 +452,7 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
else: else:
if "type" in nn.attrib: if "type" in nn.attrib:
if nn.attrib['type'] in all_defined_graphic_types:
c_t_in_graphic[nn.attrib['type']].append( [ int(float(vv.attrib['x'])) , int(float(vv.attrib['y'])) ] ) c_t_in_graphic[nn.attrib['type']].append( [ int(float(vv.attrib['x'])) , int(float(vv.attrib['y'])) ] )
sumi+=1 sumi+=1

Loading…
Cancel
Save