Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save FrancescoSaverioZuppichini/a1f84c9acff164c8e162b32a6164231e to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/a1f84c9acff164c8e162b32a6164231e to your computer and use it in GitHub Desktop.
Test
nms_indices = nms(bboxes, scores, .45)
plot_bboxes(img,
bboxes[nms_indices],
colors=["yellow" if el.item() == 0 else "blue" for el in labels[nms_indices]],
labels=["head" if el.item() == 0 else "mic" for el in labels[nms_indices]]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment