From 74311d45e63fb735ad41230dad957ffa7e9fe456 Mon Sep 17 00:00:00 2001 From: Vladisalv Sovrasov Date: Wed, 26 Feb 2025 01:35:56 +0900 Subject: [PATCH 1/3] Convert labels to int in det result --- src/python/model_api/models/result/detection.py | 2 +- tests/python/unit/results/test_det_result.py | 15 +++++++++++++++ 2 files changed, 16 insertions(+), 1 deletion(-) create mode 100644 tests/python/unit/results/test_det_result.py diff --git a/src/python/model_api/models/result/detection.py b/src/python/model_api/models/result/detection.py index f461f610..f98bbd2c 100644 --- a/src/python/model_api/models/result/detection.py +++ b/src/python/model_api/models/result/detection.py @@ -34,7 +34,7 @@ def __init__( ): super().__init__() self._bboxes = bboxes - self._labels = labels + self._labels = labels.astype(np.int32) self._scores = scores if scores is not None else np.zeros(len(bboxes)) self._label_names = ["#"] * len(bboxes) if label_names is None else label_names self._saliency_map = saliency_map diff --git a/tests/python/unit/results/test_det_result.py b/tests/python/unit/results/test_det_result.py new file mode 100644 index 00000000..ec2546e4 --- /dev/null +++ b/tests/python/unit/results/test_det_result.py @@ -0,0 +1,15 @@ +# +# Copyright (C) 2025 Intel Corporation +# SPDX-License-Identifier: Apache-2.0 +# + +import numpy as np +from model_api.models.result import DetectionResult + + +def test_cls_result(): + tst_vector = np.array([1, 2, 3, 4], dtype=np.float32) + det_result = DetectionResult(tst_vector, tst_vector, tst_vector, ["a"], tst_vector, tst_vector) + + assert det_result.labels.dtype == np.int32 + assert len(det_result.label_names) == 1 From 0fcbb82572f007463c3492a60663e07d71706de7 Mon Sep 17 00:00:00 2001 From: Vladisalv Sovrasov Date: Wed, 26 Feb 2025 01:53:44 +0900 Subject: [PATCH 2/3] Fix linter --- tests/python/unit/results/test_det_result.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/python/unit/results/test_det_result.py b/tests/python/unit/results/test_det_result.py index ec2546e4..cec500e2 100644 --- a/tests/python/unit/results/test_det_result.py +++ b/tests/python/unit/results/test_det_result.py @@ -9,7 +9,9 @@ def test_cls_result(): tst_vector = np.array([1, 2, 3, 4], dtype=np.float32) - det_result = DetectionResult(tst_vector, tst_vector, tst_vector, ["a"], tst_vector, tst_vector) + det_result = DetectionResult( + tst_vector, tst_vector, tst_vector, ["a"], tst_vector, tst_vector + ) assert det_result.labels.dtype == np.int32 assert len(det_result.label_names) == 1 From 2d4fc8575e6fd74ac280abbdf216d85061780a96 Mon Sep 17 00:00:00 2001 From: Vladisalv Sovrasov Date: Wed, 26 Feb 2025 01:54:13 +0900 Subject: [PATCH 3/3] Bump minor version --- src/python/pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/python/pyproject.toml b/src/python/pyproject.toml index 20e12347..69005c15 100644 --- a/src/python/pyproject.toml +++ b/src/python/pyproject.toml @@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta" [project] name = "openvino_model_api" -version = "0.3.0" +version = "0.3.0.1" requires-python = ">=3.9" authors = [ {name = "Intel(R) Corporation"},