Skip to content

Commit 75d5a9c

Browse files
authored
Merge pull request #3411 from ivikhrev/fix
Fix ultra-lightweight-face-detection postprocessing
2 parents 8841555 + 1024cdc commit 75d5a9c

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)