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: prevent crash when printspace is missing in xmls used for label generation

Browse source
This commit is contained in:
vahidrezanezhad 2026-01-26 17:30:26 +01:00
parent 6ae244bf9b
commit 33f6a231bc
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -737,9 +737,12 @@ def get_images_of_ground_truth(gt_list, dir_in, output_dir, output_type, config_
cnt_size = np.array([cv2.contourArea(contours[j]) for j in range(len(contours))]) cnt_size = np.array([cv2.contourArea(contours[j]) for j in range(len(contours))])
cnt = contours[np.argmax(cnt_size)] try:
cnt = contours[np.argmax(cnt_size)]
x, y, w, h = cv2.boundingRect(cnt)
except:
x, y , w, h = 0, 0, x_len, y_len
x, y, w, h = cv2.boundingRect(cnt)
bb_xywh = [x, y, w, h] bb_xywh = [x, y, w, h]