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

Created October 26, 2022 07:57

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/FrancescoSaverioZuppichini/f6574e6a9f5bae5c2dcb2ff77e102aa9.js"></script>
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