From 1cd66654cb487199711684969a3bee6a188f6506 Mon Sep 17 00:00:00 2001
From: Wang Xin <xinwang614@gmail.com>
Date: Thu, 11 May 2023 16:13:14 +0800
Subject: [PATCH] bug fixed for output_format == bbox (#22)

---
 src/labelme2yolo/l2y.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/labelme2yolo/l2y.py b/src/labelme2yolo/l2y.py
index 5e18a75..a117397 100644
--- a/src/labelme2yolo/l2y.py
+++ b/src/labelme2yolo/l2y.py
@@ -102,7 +102,13 @@ def extend_point_list(point_list, out_format="polygon"):
     if (out_format == "polygon"):
         return np.array([xmin, ymin, xmax, ymin, xmax, ymax, xmin, ymax])
     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):