NN Archive - Parsers Support (#127)

luxonis · Jul 5, 2024 · 30cf4d3 · 30cf4d3
1 parent b688a92
commit 30cf4d3
Show file tree

Hide file tree

Showing 10 changed files with 522 additions and 547 deletions.
diff --git a/luxonis_ml/nn_archive/config_building_blocks/__init__.py b/luxonis_ml/nn_archive/config_building_blocks/__init__.py
@@ -1,11 +1,6 @@
 from .base_models import (
     Head,
-    HeadClassification,
-    HeadObjectDetection,
-    HeadObjectDetectionSSD,
-    HeadSegmentation,
     HeadType,
-    HeadYOLO,
     Input,
     Metadata,
     Output,
@@ -21,11 +16,6 @@
 __all__ = [
     "Head",
     "HeadType",
-    "HeadSegmentation",
-    "HeadClassification",
-    "HeadObjectDetectionSSD",
-    "HeadObjectDetection",
-    "HeadYOLO",
     "Input",
     "PreprocessingBlock",
     "Output",

diff --git a/luxonis_ml/nn_archive/config_building_blocks/base_models/__init__.py b/luxonis_ml/nn_archive/config_building_blocks/base_models/__init__.py
@@ -1,11 +1,6 @@
 from .head import (
     Head,
-    HeadClassification,
-    HeadObjectDetection,
-    HeadObjectDetectionSSD,
-    HeadSegmentation,
     HeadType,
-    HeadYOLO,
 )
 from .input import Input, PreprocessingBlock
 from .metadata import Metadata
@@ -14,11 +9,6 @@
 __all__ = [
     "Head",
     "HeadType",
-    "HeadSegmentation",
-    "HeadClassification",
-    "HeadObjectDetection",
-    "HeadObjectDetectionSSD",
-    "HeadYOLO",
     "Input",
     "PreprocessingBlock",
     "Output",