qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-01-29 13:46:58 +01:00
Code Issues Projects Releases Wiki Activity

Fix label generation by selecting largest contour when erosion splits shapes

Browse source
This commit is contained in:
vahidrezanezhad 2026-01-26 13:36:24 +01:00
parent 49261fa99b
commit c8240905a8
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/eynollah/training/gt_gen_utils.py
View file

@ -231,7 +231,12 @@ def update_region_contours(co_text, img_boundary, erosion_rate, dilation_rate, y
con_eroded = return_contours_of_interested_region(img_boundary_in,pixel, min_size )
try:
co_text_eroded.append(con_eroded[0])
if len(con_eroded)>1:
cnt_size = np.array([cv2.contourArea(con_eroded[j]) for j in range(len(con_eroded))])
cnt = contours[np.argmax(cnt_size)]
co_text_eroded.append(cnt)
else:
co_text_eroded.append(con_eroded[0])
except:
co_text_eroded.append(con)