From 3996ae0d55730a87bbfd73c959885a8d5fbe5713 Mon Sep 17 00:00:00 2001 From: jkbmrz <74824974+jkbmrz@users.noreply.github.com> Date: Thu, 11 Jul 2024 16:29:05 +0200 Subject: [PATCH 1/2] Fix and Enhance Head Metadata Validation (#132) --- .../nn_archive/config_building_blocks/base_models/head.py | 2 +- .../config_building_blocks/base_models/head_metadata.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) 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", From a5647a5579e867f2a630eefdf7562814c88f865f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20Kozlovsk=C3=BD?= Date: Thu, 11 Jul 2024 17:00:11 +0200 Subject: [PATCH 2/2] Minor Version Bump (v0.2.3) (#133) --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"