🗑️ Remove Object Detection 3d (#4223)

Signed-off-by: Ashwin Vaidya <[email protected]>

Author: ashwinvaidya17

.github/workflows/pre_merge.yaml

@@ -139,7 +139,6 @@ jobs:
           - task: "anomaly_detection"
           - task: "anomaly_segmentation"
           - task: "keypoint_detection"
-          - task: "object_detection_3d"
     name: Integration-Test-${{ matrix.task }}-py310
     steps:
       - name: Checkout repository

CHANGELOG.md

@@ -75,6 +75,11 @@ All notable changes to this project will be documented in this file.
 - Revert h-cls head to linear one
   (<https://github.com/openvinotoolkit/training_extensions/pull/4221>)
 
+### Removed
+
+- Remove object detection 3D task
+  (<https://github.com/openvinotoolkit/training_extensions/pull/4223>)
+
 ## \[2.2.2\]
 
 ### Bug fixes

src/otx/algo/__init__.py

@@ -19,7 +19,6 @@
     "diffusion",
     "keypoint_detection",
     "modules",
-    "object_detection_3d",
     "plugins",
     "samplers",
     "segmentation",

src/otx/algo/common/layers/position_embed.py

@@ -8,7 +8,6 @@
 import math
 
 import torch
-from otx.algo.object_detection_3d.utils.utils import NestedTensor
 from torch import nn
 
 
@@ -41,14 +40,11 @@ def __init__(
             scale = 2 * math.pi
         self.scale = scale
 
-    def forward(self, tensor_list: NestedTensor | torch.Tensor) -> torch.Tensor:
+    def forward(self, tensor_list: torch.Tensor) -> torch.Tensor:
         """Forward function for PositionEmbeddingSine module."""
         if isinstance(tensor_list, torch.Tensor):
             x = tensor_list
             mask = torch.zeros((x.size(0), x.size(2), x.size(3)), device=x.device, dtype=torch.bool)
-        elif isinstance(tensor_list, NestedTensor):
-            x = tensor_list.tensors
-            mask = tensor_list.mask
         else:
             msg = f"Unrecognized type {type(tensor_list)}"
             raise TypeError(msg)