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 index 5ef1d549..9084c28c 100644 --- 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 index e3581d71..f06ac2b2 100644 --- 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 index fab258d0..cc581eaa 100644 --- 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"