fix(models/utils.py): refactor nms() implmentation, same with torchvi… - Githubissues

triple-Mu / YOLOv8-TensorRT

YOLOv8 using TensorRT accelerate !

MIT License

1.32k stars 228 forks source link

fix(models/utils.py): refactor nms() implmentation, same with torchvi… #192

Open zjykzj opened 8 months ago

zjykzj commented 8 months ago

Fix this: nms using numpy is not equivalent to torchvision.ops.nms

See #183