Skip to content

Commit 9232d6f

Browse files
author
Evgeny Tsykunov
authored
Fix OV batch inference (saliency map generation) (#1965)
Fix OV batch SM infer
1 parent a734443 commit 9232d6f

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

otx/algorithms/classification/tasks/openvino.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ def explain(
273273
dataset_item.append_labels(item_labels)
274274
add_saliency_maps_to_dataset_item(
275275
dataset_item=dataset_item,
276-
saliency_map=np.copy(saliency_map),
276+
saliency_map=saliency_map,
277277
model=self.model,
278278
labels=self.task_environment.get_labels(),
279279
predicted_scored_labels=item_labels,

otx/algorithms/detection/tasks/openvino.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@ def explain(
465465

466466
add_saliency_maps_to_dataset_item(
467467
dataset_item=dataset_item,
468-
saliency_map=np.copy(saliency_map),
468+
saliency_map=saliency_map,
469469
model=self.model,
470470
labels=labels,
471471
predicted_scored_labels=predicted_scored_labels,

otx/api/entities/result_media.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def __init__(
8282
self.annotation_scene = annotation_scene
8383
self.roi = Annotation(Rectangle.generate_full_box(), labels=[]) if roi is None else roi
8484
self.label = label
85-
self.numpy = numpy
85+
self.numpy = np.copy(numpy)
8686

8787
def __repr__(self):
8888
"""Returns a string with all the attributes of the ResultMediaEntity."""

0 commit comments

Comments
 (0)