remove support for ObjectDetectionSSD

luxonis · Mar 14, 2024 · 87bd9b2 · 87bd9b2
1 parent 3c0ddc3
commit 87bd9b2
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/luxonis_train/core/archiver.py b/luxonis_train/core/archiver.py
@@ -269,9 +269,6 @@ def _get_head_specific_parameters(
             # head_outputs["n_keypoints"] # TODO: implement
             # head_outputs["n_prototypes"] # TODO: implement
             # head_outputs["prototype_output_name"] # TODO: implement
-        elif head_name == "ObjectDetectionSSD":
-            raise NotImplementedError
-            # head_outputs["anchors"] # TODO: implement
         elif head_name == "SegmentationHead":
             parameters["is_softmax"] = self._is_softmax(executable_path)  # TODO: test
         elif head_name == "BiSeNetHead":
@@ -324,8 +321,6 @@ def _get_head_outputs(self, head_name) -> dict:
             head_outputs["yolo_outputs"] = [
                 output_dict["name"] for output_dict in self.outputs
             ]  # TODO: test
-        elif head_name == "ObjectDetectionSSD":
-            raise NotImplementedError  # TODO: boxes, scores
         elif head_name == "SegmentationHead":
             raise NotImplementedError  # TODO: predictions
         elif head_name == "BiSeNetHead":