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

Created October 26, 2022 07:57

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

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/FrancescoSaverioZuppichini/78bd337ee55363b18ddd71827b35c96b.js"></script>
Save FrancescoSaverioZuppichini/78bd337ee55363b18ddd71827b35c96b to your computer and use it in GitHub Desktop.

Download ZIP

Test

Raw

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

	%%timeit
	nms(bboxes + labels[..., None], scores, .45)

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