Don't show segmentation labels for unused segmentations

mlavik1 · Jan 29, 2025 · ce816fb · ce816fb
1 parent a441daf
commit ce816fb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Assets/Scripts/Segmentation/SegmentationBuilder.cs b/Assets/Scripts/Segmentation/SegmentationBuilder.cs
@@ -35,6 +35,7 @@ public static List<SegmentationLabel> BuildSegmentations(VolumeDataset dataset)
             List<SegmentationLabel> result = new List<SegmentationLabel>();
             int minSegmentationId = int.MaxValue;
             int maxSegmentationId = int.MinValue;
+            HashSet<int> usedSegmentationIds = new HashSet<int>();
 
             for (int i = 0; i < dataset.data.Length; i++)
             {
@@ -43,13 +44,16 @@ public static List<SegmentationLabel> BuildSegmentations(VolumeDataset dataset)
                 {
                     minSegmentationId = Mathf.Min(minSegmentationId, value);
                     maxSegmentationId = Mathf.Max(maxSegmentationId, value);
+                    usedSegmentationIds.Add(value);
                 }
             }
 
             bool multiLabel = maxSegmentationId - minSegmentationId > 1;
 
             for (int segmentationId = minSegmentationId; segmentationId <= maxSegmentationId; segmentationId++)
             {
+                if (!usedSegmentationIds.Contains(segmentationId))
+                    continue;
                 SegmentationLabel segmentationLabel = new SegmentationLabel();
                 segmentationLabel.id = segmentationId;
                 segmentationLabel.name = dataset.datasetName;