Skip to content

Instantly share code, notes, and snippets.

@FrancescoSaverioZuppichini

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

Created October 26, 2022 07:57
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save FrancescoSaverioZuppichini/f6574e6a9f5bae5c2dcb2ff77e102aa9 to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/f6574e6a9f5bae5c2dcb2ff77e102aa9 to your computer and use it in GitHub Desktop.
Download ZIP
Test
Raw
non-max-suppression-in-pytorch-21.py
overlapped = torch.nonzero(iou > iou_threshold)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment