FrancescoSaverioZuppichini/non-max-suppression-in-pytorch-5.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/0379c76ee70dd9422c065252ee065fc8.js"></script>
Save FrancescoSaverioZuppichini/0379c76ee70dd9422c065252ee065fc8 to your computer and use it in GitHub Desktop.

Download ZIP

Test

Raw

non-max-suppression-in-pytorch-5.py

	scores = torch.tensor([
	0.98, 0.85, 0.5, 0.2, # for head
	1, 0.92, 0.3, 0.1 # for mic
	])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment