Skip to content

Instantly share code, notes, and snippets.

@FrancescoSaverioZuppichini

FrancescoSaverioZuppichini/non-max-suppression-in-pytorch-18.txt

Last active October 27, 2022 08:31
Show Gist options
  • Save FrancescoSaverioZuppichini/17cdce6577141912a77a043bfab4bd6e to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/17cdce6577141912a77a043bfab4bd6e to your computer and use it in GitHub Desktop.
Download ZIP
Test
Raw
non-max-suppression-in-pytorch-18.txt
iou = box_iou(bbox[None,...], (bboxes[order[i + 1:]]) * keep[i + 1:][...,None])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment