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

fixing artificial class bug

Browse Source
pull/18/head
vahidrezanezhad 4 months ago
parent c502e67c14
commit 5f456cf508
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File

22
gt_gen_utils.py
Unescape Escape View File

@ -8,6 +8,7 @@ from tqdm import tqdm
import cv2 import cv2
from shapely import geometry from shapely import geometry
from pathlib import Path from pathlib import Path
import matplotlib.pyplot as plt
KERNEL = np.ones((5, 5), np.uint8) KERNEL = np.ones((5, 5), np.uint8)
@ -83,9 +84,13 @@ def return_contours_of_interested_region(region_pre_p, pixel, min_area=0.0002):
ret, thresh = cv2.threshold(imgray, 0, 255, 0) ret, thresh = cv2.threshold(imgray, 0, 255, 0)
contours_imgs, hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE) 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 = 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 return contours_imgs
def update_region_contours(co_text, img_boundary, erosion_rate, dilation_rate, y_len, x_len, dilation_early=None): 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 pixel = 1
min_size = 0 min_size = 0
img_boundary_in = img_boundary_in.astype("uint8")
con_eroded = return_contours_of_interested_region(img_boundary_in,pixel, min_size ) con_eroded = return_contours_of_interested_region(img_boundary_in,pixel, min_size )
try: #try:
co_text_eroded.append(con_eroded[0]) co_text_eroded.append(con_eroded[0])
except: #except:
co_text_eroded.append(con) #co_text_eroded.append(con)
img_boundary_in_dilated = cv2.dilate(img_boundary_in[:,:], KERNEL, iterations=dilation_rate) 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: if "artificial_class_label" in keys:
img_boundary = np.zeros((y_len, x_len)) img_boundary = np.zeros((y_len, x_len))
erosion_rate = 0#1 erosion_rate = 0#1
dilation_rate = 3 dilation_rate = 2
dilation_early = 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 ) 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 )

Write Preview
Loading…
Cancel
Save