diff --git a/src/otx/core/data/dataset/segmentation.py b/src/otx/core/data/dataset/segmentation.py index 5672989e7f..8f9d8f8fe7 100644 --- a/src/otx/core/data/dataset/segmentation.py +++ b/src/otx/core/data/dataset/segmentation.py @@ -184,6 +184,7 @@ def __init__( if self.has_polygons: # insert background class at index 0 since polygons represent only objects self.label_info.label_names.insert(0, "otx_background_lbl") + self.label_info.label_ids.insert(0, "None") self.label_info = SegLabelInfo( label_names=self.label_info.label_names,