Skip to content

Commit 848de1a

Browse files
authored
Fix OV wrapper backward compatibility for IS models (#2143)
1 parent 1730b79 commit 848de1a

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

otx/algorithms/detection/adapters/openvino/model_wrappers/openvino_models.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,13 +67,16 @@ def postprocess(self, outputs, meta):
6767
# pylint: disable-msg=too-many-locals
6868
# FIXME: here, batch dim of IR must be 1
6969
boxes = outputs[self.output_blob_name["boxes"]]
70-
boxes = boxes.squeeze(0)
70+
if boxes.shape[0] == 1:
71+
boxes = boxes.squeeze(0)
7172
assert boxes.ndim == 2
7273
masks = outputs[self.output_blob_name["masks"]]
73-
masks = masks.squeeze(0)
74+
if masks.shape[0] == 1:
75+
masks = masks.squeeze(0)
7476
assert masks.ndim == 3
7577
classes = outputs[self.output_blob_name["labels"]].astype(np.uint32)
76-
classes = classes.squeeze(0)
78+
if classes.shape[0] == 1:
79+
classes = classes.squeeze(0)
7780
assert classes.ndim == 1
7881
if self.is_segmentoly:
7982
scores = outputs[self.output_blob_name["scores"]]

0 commit comments

Comments
 (0)