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

fixing artificial class bug

Browse source
This commit is contained in:
vahidrezanezhad 2024-08-28 17:34:06 +02:00
parent c502e67c14
commit 5f456cf508
1 changed files with 15 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

20
gt_gen_utils.py
View file

@ -8,6 +8,7 @@ from tqdm import tqdm
import cv2
from shapely import geometry
from pathlib import Path
import matplotlib.pyplot as plt
KERNEL = np.ones((5, 5), np.uint8)
@ -84,8 +85,12 @@ def return_contours_of_interested_region(region_pre_p, pixel, min_area=0.0002):
contours_imgs, hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
#print(len(contours_imgs), hierarchy)
contours_imgs = return_parent_contours(contours_imgs, hierarchy)
contours_imgs = filter_contours_area_of_image_tables(thresh, contours_imgs, hierarchy, max_area=1, min_area=min_area)
#print(len(contours_imgs), "iki")
#contours_imgs = filter_contours_area_of_image_tables(thresh, contours_imgs, hierarchy, max_area=1, min_area=min_area)
return contours_imgs
def update_region_contours(co_text, img_boundary, erosion_rate, dilation_rate, y_len, x_len, dilation_early=None):
@ -103,12 +108,15 @@ def update_region_contours(co_text, img_boundary, erosion_rate, dilation_rate, y
pixel = 1
min_size = 0
img_boundary_in = img_boundary_in.astype("uint8")
con_eroded = return_contours_of_interested_region(img_boundary_in,pixel, min_size )
try:
#try:
co_text_eroded.append(con_eroded[0])
except:
co_text_eroded.append(con)
#except:
#co_text_eroded.append(con)
img_boundary_in_dilated = cv2.dilate(img_boundary_in[:,:], KERNEL, iterations=dilation_rate)
@ -262,8 +270,8 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
if "artificial_class_label" in keys:
img_boundary = np.zeros((y_len, x_len))
erosion_rate = 0#1
dilation_rate = 3
dilation_early = 2
dilation_rate = 2
dilation_early = 1
co_use_case, img_boundary = update_region_contours(co_use_case, img_boundary, erosion_rate, dilation_rate, y_len, x_len, dilation_early=dilation_early )