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

laugh12321 / TensorRT-YOLO

🚀 你的YOLO部署神器。TensorRT Plugin、CUDA Kernel、CUDA Graphs三管齐下，享受闪电般的推理速度。| Your YOLO Deployment Powerhouse. With the synergy of TensorRT Plugins, CUDA Kernels, and CUDA Graphs, experience lightning-fast inference speeds.

https://github.com/laugh12321/TensorRT-YOLO

GNU General Public License v3.0

720 stars 81 forks source link

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

Closed liautumn closed 7 months ago

liautumn commented 7 months ago

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

laugh12321 commented 7 months ago

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

liautumn commented 7 months ago

好的，感谢

laugh12321 commented 7 months ago

没事