Skip to content

Commit eeb1c2f

Browse files
committed
get coords from keep only
1 parent 8841555 commit eeb1c2f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

demos/common/python/openvino/model_zoo/model_api/models/ultra_lightweight_face_detection.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,5 +73,4 @@ def _parse_outputs(self, outputs, meta):
7373

7474
keep = nms(x_mins, y_mins, x_maxs, y_maxs, filtered_score, self.iou_threshold)
7575

76-
filtered_score = filtered_score[keep]
77-
return [Detection(*det, 0) for det in zip(x_mins, y_mins, x_maxs, y_maxs, filtered_score)]
76+
return [Detection(*det, 0) for det in zip(x_mins[keep], y_mins[keep], x_maxs[keep], y_maxs[keep], filtered_score[keep])]

0 commit comments

Comments
 (0)