Fix: LuxonisDataset Converter - bbox computation (#70)

Fix: computation of x, y, w, h
luxonis · Nov 3, 2024 · 77be96e · 77be96e
1 parent 749696d
commit 77be96e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/datadreamer/utils/luxonis_dataset_converter.py b/datadreamer/utils/luxonis_dataset_converter.py
@@ -109,18 +109,18 @@ def dataset_generator():
                 if "boxes" in data[image_path]:
                     boxes = data[image_path]["boxes"]
                     for box, label in zip(boxes, labels):
-                        x, y, w, h = box[0], box[1], box[2] - box[0], box[3] - box[1]
-                        x = max(0, x)
-                        y = max(0, y)
+                        x, y = max(0, box[0] / width), max(0, box[1] / height)
+                        w = min(box[2] / width - x, 1 - x)
+                        h = min(box[3] / height - y, 1 - y)
                         yield {
                             "file": image_full_path,
                             "annotation": {
                                 "class": class_names[label],
                                 "type": "boundingbox",
-                                "x": x / width,
-                                "y": y / height,
-                                "w": w / width,
-                                "h": h / height,
+                                "x": x,
+                                "y": y,
+                                "w": w,
+                                "h": h,
                             },
                         }
 

diff --git a/media/coverage_badge.svg b/media/coverage_badge.svg