search

triple-Mu / YOLOv8-TensorRT

YOLOv8 using TensorRT accelerate !
MIT License
1.29k stars 223 forks source link

how to add conf-thres，when use an engine file to predict #234

Open uflexnihao opened 1 month ago

uflexnihao commented 1 month ago

use YOLOv8-TensorRT ,I generated engine and it works. I want to know ，use the engine how can I set a new conf-thres.

triple-Mu commented 1 month ago

use YOLOv8-TensorRT ,I generated engine and it works. I want to know ，use the engine how can I set a new conf-thres.

The end2end model has already contain the conf-thres in onnx. If you want to customize the conf-thres, you should try normal https://github.com/triple-Mu/YOLOv8-TensorRT/blob/main/docs/Normal.md