compare shape instead of annotation for appending labels

openvinotoolkit · May 12, 2022 · 6e534c6 · 6e534c6
1 parent 9e916f1
commit 6e534c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ote_sdk/ote_sdk/entities/dataset_item.py b/ote_sdk/ote_sdk/entities/dataset_item.py
@@ -401,7 +401,7 @@ def append_labels(self, labels: List[ScoredLabel]):
 
         roi_annotation = None
         for annotation in self.annotation_scene.annotations:
-            if annotation == self.roi:
+            if annotation.shape == self.roi.shape:
                 roi_annotation = annotation
                 break