Skip to content

Instantly share code, notes, and snippets.

@FrancescoSaverioZuppichini

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

Created October 26, 2022 07:57
Show Gist options
  • Save FrancescoSaverioZuppichini/2b15163b5f0ba1d4e0a4171e8001d6c6 to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/2b15163b5f0ba1d4e0a4171e8001d6c6 to your computer and use it in GitHub Desktop.
Download ZIP
Test
Raw
non-max-suppression-in-pytorch-2.py
original_bboxes = torch.tensor([
# head
[ 565, 73, 862, 373],
# mic
[807, 309, 865, 434]
]).float()
w, h = img.size
# we need them in range [0, 1]
original_bboxes[...,0] /= h
original_bboxes[...,1] /= w
original_bboxes[...,2] /= h
original_bboxes[...,3] /= w
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment