Skip to content

Commit 74311d4

Browse files
committed
Convert labels to int in det result
1 parent 4251bd4 commit 74311d4

File tree

2 files changed

+16
-1
lines changed

2 files changed

+16
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
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
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
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(tst_vector, tst_vector, tst_vector, ["a"], tst_vector, tst_vector)
13+
14+
assert det_result.labels.dtype == np.int32
15+
assert len(det_result.label_names) == 1

0 commit comments

Comments
 (0)