Skip to content

Commit ad6ccda

Browse files
djdamelndaniil-lyakhov
authored andcommitted
🗑 Remove normal labels before evaluation (openvinotoolkit#1219)
1 parent abae2c6 commit ad6ccda

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

‎ote_sdk/ote_sdk/utils/dataset_utils.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -152,8 +152,9 @@ def split_local_global_resultset(
152152
:param resultset: Input result set
153153
:return: Globally annotated result set, locally annotated result set
154154
"""
155-
global_gt_dataset, local_gt_dataset = split_local_global_dataset(
156-
resultset.ground_truth_dataset
155+
global_gt_dataset = get_global_subset(resultset.ground_truth_dataset)
156+
local_gt_dataset = get_local_subset(
157+
resultset.ground_truth_dataset, include_normal=False
157158
)
158159
local_idx = get_fully_annotated_idx(resultset.ground_truth_dataset)
159160
global_pred_dataset = get_global_subset(resultset.prediction_dataset)

0 commit comments

Comments
 (0)