Skip to content

Commit 9157ebb

Browse files
authoredFeb 26, 2025··
Convert labels to int in det result (#272)
* Convert labels to int in det result * Fix linter * Bump minor version
1 parent 4251bd4 commit 9157ebb

File tree

3 files changed

+19
-2
lines changed

3 files changed

+19
-2
lines changed
 

‎src/python/model_api/models/result/detection.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def __init__(
3434
):
3535
super().__init__()
3636
self._bboxes = bboxes
37-
self._labels = labels
37+
self._labels = labels.astype(np.int32)
3838
self._scores = scores if scores is not None else np.zeros(len(bboxes))
3939
self._label_names = ["#"] * len(bboxes) if label_names is None else label_names
4040
self._saliency_map = saliency_map

‎src/python/pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
99

1010
[project]
1111
name = "openvino_model_api"
12-
version = "0.3.0"
12+
version = "0.3.0.1"
1313
requires-python = ">=3.9"
1414
authors = [
1515
{name = "Intel(R) Corporation"},
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
#
2+
# Copyright (C) 2025 Intel Corporation
3+
# SPDX-License-Identifier: Apache-2.0
4+
#
5+
6+
import numpy as np
7+
from model_api.models.result import DetectionResult
8+
9+
10+
def test_cls_result():
11+
tst_vector = np.array([1, 2, 3, 4], dtype=np.float32)
12+
det_result = DetectionResult(
13+
tst_vector, tst_vector, tst_vector, ["a"], tst_vector, tst_vector
14+
)
15+
16+
assert det_result.labels.dtype == np.int32
17+
assert len(det_result.label_names) == 1

0 commit comments

Comments
 (0)
Please sign in to comment.