chore: remove duplicated SegmentationData2D (#24)

Signed-off-by: ktro2828 <[email protected]>

t4_devkit/viewer/rendering_data/box.py

@@ -7,7 +7,7 @@
 
 if TYPE_CHECKING:
     from t4_devkit.dataclass import Box2D, Box3D
-    from t4_devkit.typing import NDArrayU8, RoiType, SizeType, TranslationType, VelocityType
+    from t4_devkit.typing import RoiType, SizeType, TranslationType, VelocityType
 
 __all__ = ["BoxData3D", "BoxData2D"]
 
@@ -107,39 +107,3 @@ def as_boxes2d(self) -> rr.Boxes2D:
             labels=labels,
             class_ids=self._class_ids,
         )
-
-
-class Segmentation2D:
-    def __init__(self) -> None:
-        self._masks: list[NDArrayU8] = []
-        self._class_ids: list[int] = []
-
-        self._size: tuple[int, int] = None  # (height, width)
-
-    def append(self, mask: NDArrayU8, class_id: int) -> None:
-        """Append a segmentation mask and its class ID.
-
-        Args:
-            mask (NDArrayU8): Mask image, in the shape of (H, W).
-            class_id (int): Class ID.
-
-        Raises:
-            ValueError: _description_
-        """
-        if self._size is None:
-            self._size = mask.shape
-        else:
-            if self._size != mask.shape:
-                raise ValueError(
-                    f"All masks must be the same size. Expected: {self._size}, but got {mask.shape}"
-                )
-        self._masks.append(mask)
-        self._class_ids.append(class_id)
-
-    def as_segmentation_image(self) -> rr.SegmentationImage:
-        image = np.zeros(self._size, dtype=np.uint8)
-
-        for mask, class_id in zip(self._masks, self._class_ids, strict=True):
-            image[mask == 1] == class_id
-
-        return rr.SegmentationImage(data=image)