diff --git a/README.md b/README.md
index f08632d..9fd4745 100644
--- 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
index 64f4968..8462ea1 100644
--- 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
index 3afd902..2ab6b13 100644
--- 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
\ No newline at end of file