yolov5修改nms
时间: 2023-11-10 21:03:20 浏览: 289
基于yolov5的旋转目标检测yolov5-obb-master.zip
对于修改 YOLOv5 中的 NMS,可以在 `detect.py` 文件中的 `non_max_suppression` 函数中进行修改。具体来说,可以调整 `iou_threshold` 参数来改变 NMS 的阈值,也可以修改 `max_det` 参数来控制每个类别最多检测出多少个目标。
另外,如果需要修改 NMS 的具体实现方式,可以参考 `utils/general.py` 文件中的 `non_max_suppression` 函数,自行实现新的 NMS 算法。
阅读全文