Skip to content

Commit 4d379c3

Browse files
committed
Convert labels to int in det result
1 parent 4251bd4 commit 4d379c3

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-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: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
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+
tst_vector = np.array([1, 2, 3, 4], dtype=np.float32)
13+
det_result = DetectionResult(tst_vector, tst_vector, tst_vector, ["a"], tst_vector, tst_vector)
14+
15+
assert det_result.labels.dtype == np.int32
16+
assert len(det_result.label_names) == 1

0 commit comments

Comments
 (0)