mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 11:50:04 +02:00
just defined graphic region types can be extracted as label
This commit is contained in:
parent
b1d971a200
commit
dc356a5f42
1 changed files with 6 additions and 3 deletions
|
@ -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":[]}
|
||||
all_defined_textregion_types = list(co_text.keys())
|
||||
co_graphic = {"handwritten-annotation":[], "decoration":[], "stamp":[], "signature":[]}
|
||||
all_defined_graphic_types = list(co_graphic.keys())
|
||||
co_sep=[]
|
||||
co_img=[]
|
||||
co_table=[]
|
||||
|
@ -425,7 +426,8 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
|
|||
|
||||
else:
|
||||
if "type" in nn.attrib:
|
||||
c_t_in_graphic[nn.attrib['type']].append( np.array( [ [ int(x.split(',')[0]) , int(x.split(',')[1]) ] for x in p_h] ) )
|
||||
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] ) )
|
||||
|
||||
break
|
||||
else:
|
||||
|
@ -450,8 +452,9 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
|
|||
|
||||
else:
|
||||
if "type" in nn.attrib:
|
||||
c_t_in_graphic[nn.attrib['type']].append( [ int(float(vv.attrib['x'])) , int(float(vv.attrib['y'])) ] )
|
||||
sumi+=1
|
||||
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'])) ] )
|
||||
sumi+=1
|
||||
|
||||
elif vv.tag!=link+'Point' and sumi>=1:
|
||||
break
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue