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

replacement in a list done correctly

Browse source
This commit is contained in:
vahidrezanezhad 2024-06-06 14:38:29 +02:00
parent 821290c464
commit b9cbc0edb7
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
gt_gen_utils.py
View file

@ -636,7 +636,7 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
if 'textregions' in keys:
if 'rest_as_paragraph' in types_text:
types_text[types_text=='rest_as_paragraph'] = 'paragraph'
types_text = ['paragraph'if ttind=='rest_as_paragraph' else ttind for ttind in types_text]
for element_text in types_text:
if element_text == 'paragraph':
color_label = labels_rgb_color[ config_params['textregions']['rest_as_paragraph']]
@ -688,7 +688,7 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
if 'textregions' in keys:
if 'rest_as_paragraph' in types_text:
types_text[types_text=='rest_as_paragraph'] = 'paragraph'
types_text = ['paragraph'if ttind=='rest_as_paragraph' else ttind for ttind in types_text]
for element_text in types_text:
if element_text == 'paragraph':
color_label = config_params['textregions']['rest_as_paragraph']