fix: different color for different classes in the segmenetation visua…

…ization
luxonis · Oct 21, 2024 · 7879220 · 7879220
1 parent a1c6b6a
commit 7879220
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datadreamer/pipelines/generate_dataset_from_scratch.py b/datadreamer/pipelines/generate_dataset_from_scratch.py
@@ -657,7 +657,7 @@ def read_image_batch(image_batch, batch_num, batch_size):
                             mask = masks_batch[j][k]
                             x_points, y_points = zip(*mask)
 
-                            ax.fill(x_points, y_points, "blue", alpha=0.5)
+                            ax.fill(x_points, y_points, label, alpha=0.5)
 
                     labels.append(label)
                     x1, y1, x2, y2 = box