qurator-spk
/
sbb_pixelwise_segmentation
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
1
0
Fork 0
Code Issues Releases Wiki Activity

erosion rate changed

Browse Source
pull/18/head
vahidrezanezhad 5 months ago
parent 5fbe941f53
commit 59e5892f25
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File

24
gt_gen_utils.py
Unescape Escape View File

@ -577,36 +577,36 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
if "artificial_class_on_boundary" in keys: if "artificial_class_on_boundary" in keys:
img_boundary = np.zeros( (y_len,x_len) ) img_boundary = np.zeros( (y_len,x_len) )
if "paragraph" in elements_with_artificial_class: if "paragraph" in elements_with_artificial_class:
erosion_rate = 0#2 erosion_rate = 2
dilation_rate = 3#4 dilation_rate = 4
co_text['paragraph'], img_boundary = update_region_contours(co_text['paragraph'], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text['paragraph'], img_boundary = update_region_contours(co_text['paragraph'], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "drop-capital" in elements_with_artificial_class: if "drop-capital" in elements_with_artificial_class:
erosion_rate = 0 erosion_rate = 1
dilation_rate = 4 dilation_rate = 3
co_text["drop-capital"], img_boundary = update_region_contours(co_text["drop-capital"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["drop-capital"], img_boundary = update_region_contours(co_text["drop-capital"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "catch-word" in elements_with_artificial_class: if "catch-word" in elements_with_artificial_class:
erosion_rate = 0 erosion_rate = 0
dilation_rate = 2#4 dilation_rate = 3#4
co_text["catch-word"], img_boundary = update_region_contours(co_text["catch-word"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["catch-word"], img_boundary = update_region_contours(co_text["catch-word"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "page-number" in elements_with_artificial_class: if "page-number" in elements_with_artificial_class:
erosion_rate = 0 erosion_rate = 0
dilation_rate = 2#4 dilation_rate = 3#4
co_text["page-number"], img_boundary = update_region_contours(co_text["page-number"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["page-number"], img_boundary = update_region_contours(co_text["page-number"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "header" in elements_with_artificial_class: if "header" in elements_with_artificial_class:
erosion_rate = 0#1 erosion_rate = 1
dilation_rate = 3#4 dilation_rate = 4
co_text["header"], img_boundary = update_region_contours(co_text["header"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["header"], img_boundary = update_region_contours(co_text["header"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "heading" in elements_with_artificial_class: if "heading" in elements_with_artificial_class:
erosion_rate = 0#1 erosion_rate = 1
dilation_rate = 3#4 dilation_rate = 4
co_text["heading"], img_boundary = update_region_contours(co_text["heading"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["heading"], img_boundary = update_region_contours(co_text["heading"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "signature-mark" in elements_with_artificial_class: if "signature-mark" in elements_with_artificial_class:
erosion_rate = 1 erosion_rate = 1
dilation_rate = 4 dilation_rate = 4
co_text["signature-mark"], img_boundary = update_region_contours(co_text["signature-mark"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["signature-mark"], img_boundary = update_region_contours(co_text["signature-mark"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "marginalia" in elements_with_artificial_class: if "marginalia" in elements_with_artificial_class:
erosion_rate = 0#2 erosion_rate = 2
dilation_rate = 3#4 dilation_rate = 4
co_text["marginalia"], img_boundary = update_region_contours(co_text["marginalia"], img_boundary, erosion_rate, dilation_rate, y_len, x_len ) co_text["marginalia"], img_boundary = update_region_contours(co_text["marginalia"], img_boundary, erosion_rate, dilation_rate, y_len, x_len )
if "footnote" in elements_with_artificial_class: if "footnote" in elements_with_artificial_class:
erosion_rate = 0#2 erosion_rate = 0#2

Write Preview
Loading…
Cancel
Save