search

laugh12321 / TensorRT-YOLO

🚀 TensorRT-YOLO: Supports YOLOv3, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv9, YOLOv10, and PP-YOLOE using TensorRT acceleration with EfficientNMS, CUDA Kernels and CUDA Graphs!
https://github.com/laugh12321/TensorRT-YOLO
GNU General Public License v3.0
383 stars 50 forks source link

[Help]: 如何设置 NMS 参数阈值 #14

Closed liautumn closed 3 months ago

liautumn commented 3 months ago

image 这两个参数在哪设置的？好像没找到

laugh12321 commented 3 months ago

使用export.py转带NMS插件时进行设置，默认分别为0.25和0.45。 具体使用方法使用python python/export/{yolo version}/export.py -h 查看。

liautumn commented 3 months ago

好的，感谢

laugh12321 commented 3 months ago

没事