fix(pre_commit): 🎨 auto format pre-commit hooks

This commit is contained in:
pre-commit-ci[bot] 2024-02-07 08:21:35 +00:00
parent c46a2e9c08
commit 8f6887d994
1 changed files with 1 additions and 1 deletions

View File

@ -215,7 +215,7 @@ def mask_non_max_suppression_2(
for i in range(num_predictions):
if keep[i]:
condition = (ious[i] > iou_threshold) & (categories == category)
keep[i + 1:] = np.where(condition[i + 1:], False, keep[i + 1:])
keep[i + 1 :] = np.where(condition[i + 1 :], False, keep[i + 1 :])
return keep