shuoshuoshuo
/
labelme2yolo
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bug fixed for output_format == bbox (#22)

This commit is contained in:
Wang Xin 2023-05-11 16:13:14 +08:00 committed by GitHub
parent df4f172722
commit 1cd66654cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/labelme2yolo/l2y.py
View File

@ -102,7 +102,13 @@ def extend_point_list(point_list, out_format="polygon"):
if (out_format == "polygon"): if (out_format == "polygon"):
return np.array([xmin, ymin, xmax, ymin, xmax, ymax, xmin, ymax]) return np.array([xmin, ymin, xmax, ymin, xmax, ymax, xmin, ymax])
if (out_format == "bbox"): if (out_format == "bbox"):
return np.array([xmin, ymin, xmax - xmin, ymax - ymin]) x = xmin
y = ymin
w = xmax - xmin
h = ymax - ymin
x = x + w / 2
y = y + h / 2
return np.array([x, y, w, h])
def save_yolo_label(json_name, label_dir_path, target_dir, yolo_obj_list): def save_yolo_label(json_name, label_dir_path, target_dir, yolo_obj_list):