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

Author: kozlov721

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,

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

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"