torch nms
torchvision.ops.nms(c_bboxes, c_scores.view(-1), 0.45)
这里有代码:
https://github.com/IvanOfficial/dali/blob/d5d3cfe3e8fb54212f0d528d6771d3b41338e7d0/yolo_opencv.py