Update docs & luxonis dataset creation

luxonis · Oct 25, 2024 · f887910 · f887910
1 parent 335cc05
commit f887910
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -281,7 +281,7 @@ save_dir/
 {
   "image_path": {
     "boxes": [[x_min, y_min, x_max, y_max], ...],
-    "masks": [[x0, y0], [x1, y1], ...]
+    "masks": [[[x0, y0],[x1, y1],...], [[x0, y0],[x1, y1],...], ....]
     "labels": [label_index, ...]
   },
   ...

diff --git a/datadreamer/utils/luxonis_dataset_converter.py b/datadreamer/utils/luxonis_dataset_converter.py
@@ -28,6 +28,11 @@ def __init__(
         self.dataset_plugin = dataset_plugin
         self.dataset_name = dataset_name
 
+        if self.is_instance_segmentation:
+            logger.warning(
+                "Instance segmentation will be treated as semantic segmentation until the support for instance segmentation is added to Luxonis-ml."
+            )
+
     def convert(
         self,
         dataset_dir: str,

diff --git a/requirements.txt b/requirements.txt
@@ -12,6 +12,6 @@ accelerate>=0.25.0
 scipy>=1.10.0
 bitsandbytes>=0.42.0
 nltk>=3.8.1
-luxonis-ml[all]>=0.4.0
+luxonis-ml[all]>=0.4.1
 python-box>=7.1.1
 gcsfs>=2023.1.0