search

jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.74k stars 545 forks source link

Update trt_yolo.py #525

Closed shubham-shahh closed 2 years ago

shubham-shahh commented 2 years ago

I added confidence threshold as an argument since I feel its an important param for testing which should be passed in args itself