Convert coco bounding box format to [x0,y0,x1,y1] format. (#169)

lancedb · Sep 16, 2022 · 13d9290 · 13d9290
1 parent 66d2c00
commit 13d9290
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/benchmarks/parse_coco.py b/python/benchmarks/parse_coco.py
@@ -28,6 +28,10 @@ def _instances_to_df(self, split, instances_json):
         df = pd.DataFrame(instances_json["annotations"])
         df["segmentation"] = df.segmentation.apply(_convert_segmentation)
         df["iscrowd"] = df.iscrowd.astype("bool")
+        # Convert coco dataset bounding box [x,y,width,height] to [x0,y0,x1,y1] format.
+        df["bbox"] = df.bbox.apply(
+            lambda arr: [arr[0], arr[1], arr[0] + arr[2], arr[1] + arr[3]]
+        )
         category_df = pd.DataFrame(instances_json["categories"]).rename(
             {"id": "category_id"}, axis=1
         )