吴恩达在 deep learning 专项课程中指出,如果有多个分类,正确做法应该是运行多次独立的NMS,每次针对一种输出分类。
对应
torchvision.ops.boxes.batched_nms() 或 torchvision.ops.batched_nms()
不同版本pytorch接口不太一样
Pytorch实现nms (torchvision.ops.nms & torchvision.ops.boxes.batched_nms)_hxxjxw的博客-CSDN博客
multiclass nms(多类别极大值抑制)
于 2022-01-19 21:06:00 首次发布