Skip to content

Instantly share code, notes, and snippets.

@FrancescoSaverioZuppichini

FrancescoSaverioZuppichini/non-max-suppression-in-pytorch-9.py

Created October 26, 2022 07:57
Show Gist options
  • Save FrancescoSaverioZuppichini/95e13312221a66c952e137ef3042bdbc to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/95e13312221a66c952e137ef3042bdbc to your computer and use it in GitHub Desktop.
Download ZIP
Test
Raw
non-max-suppression-in-pytorch-9.py
plot_bboxes(img, bboxes,
colors=["yellow" if el.item() == 0 else "blue" for el in labels],
labels=["head" if el.item() == 0 else "mic" for el in labels]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment