Merge branch 'dev' of github.com:luxonis/luxonis-ml into dev

luxonis · Jul 11, 2024 · dda0426 · dda0426
2 parents 06345f1 + 3693214
commit dda0426
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/luxonis_ml/nn_archive/config_building_blocks/base_models/head.py b/luxonis_ml/nn_archive/config_building_blocks/base_models/head.py
@@ -31,12 +31,12 @@ class Head(BaseModel, ABC):
         description="Name of the parser responsible for processing the models output."
     )
     metadata: Union[
-        HeadMetadata,
         HeadObjectDetectionMetadata,
         HeadClassificationMetadata,
         HeadObjectDetectionSSDMetadata,
         HeadSegmentationMetadata,
         HeadYOLOMetadata,
+        HeadMetadata,
     ] = Field(description="Metadata of the parser.")
     outputs: Optional[List[str]] = Field(
         None,

diff --git a/luxonis_ml/nn_archive/config_building_blocks/base_models/head_metadata.py b/luxonis_ml/nn_archive/config_building_blocks/base_models/head_metadata.py
@@ -202,6 +202,7 @@ def validate_task_specific_fields(
         defined_params = defined_params.difference(common_fields)
 
         required_fields = {
+            "object_detection": [],
             "instance_segmentation": [
                 "n_prototypes",
                 "is_softmax",

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "luxonis-ml"
-version = "0.2.2"
+version = "0.2.3"
 description = "MLOps tools for training models for Luxonis devices"
 readme = "README.md"
 requires-python = ">=3.8"