From b5b2cc8448fb8f5e27c32d36633519081ead1c40 Mon Sep 17 00:00:00 2001 From: Vladisalv Sovrasov Date: Fri, 20 Dec 2024 01:33:17 +0900 Subject: [PATCH] Update bg label handling in sseg --- src/otx/core/data/dataset/segmentation.py | 1 + 1 file changed, 1 insertion(+) 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,