added raise error

openvinotoolkit · May 27, 2024 · 3beac1b · 3beac1b
1 parent f2a2bc8
commit 3beac1b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/otx/core/types/label.py b/src/otx/core/types/label.py
@@ -282,6 +282,14 @@ class SegLabelInfo(LabelInfo):
 
     ignore_index: int = 255
 
+    def __post_init__(self):
+        if len(self.label_names) <= 1:
+            msg = (
+                "The number of labels must be larger than 1. "
+                "Please, check dataset labels and add background label in case of binary segmentation."
+            )
+            raise ValueError(msg)
+
     @classmethod
     def from_num_classes(cls, num_classes: int) -> LabelInfo:
         """Create this object from the number of classes.