From 299bca1710762db36620368db55115826a3f1747 Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Fri, 13 Dec 2024 10:14:16 +0000 Subject: [PATCH] Update todo --- src/otx/core/data/dataset/instance_segmentation.py | 2 +- src/otx/core/data/dataset/tile.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/otx/core/data/dataset/instance_segmentation.py b/src/otx/core/data/dataset/instance_segmentation.py index 8cd8ea6c07..2f3821537b 100644 --- a/src/otx/core/data/dataset/instance_segmentation.py +++ b/src/otx/core/data/dataset/instance_segmentation.py @@ -50,7 +50,7 @@ def _get_item_impl(self, index: int) -> InstanceSegDataEntity | None: gt_bboxes, gt_labels, gt_masks, gt_polygons = [], [], [], [] - # NOTE (Eugene): + # TODO(Eugene): https://jira.devtools.intel.com/browse/CVS-159363 # Temporary solution to handle multiple annotation types. # Ideally, we should pre-filter annotations during initialization of the dataset. if Polygon.__name__ in anno_collection: # Polygon for InstSeg has higher priority diff --git a/src/otx/core/data/dataset/tile.py b/src/otx/core/data/dataset/tile.py index 502dd3530a..b94931487d 100644 --- a/src/otx/core/data/dataset/tile.py +++ b/src/otx/core/data/dataset/tile.py @@ -514,9 +514,10 @@ def _get_item_impl(self, index: int) -> TileInstSegDataEntity: # type: ignore[o gt_bboxes, gt_labels, gt_masks, gt_polygons = [], [], [], [] - # NOTE (Eugene): + # TODO(Eugene): https://jira.devtools.intel.com/browse/CVS-159363 # Temporary solution to handle multiple annotation types. # Ideally, we should pre-filter annotations during initialization of the dataset. + if Polygon.__name__ in anno_collection: # Polygon for InstSeg has higher priority for poly in anno_collection[Polygon.__name__]: bbox = Bbox(*poly.get_bbox()).points