NN Archive - Parsers Support (#127)

Author: kkeroo

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",

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",