What are topk and keeptopk parameters?

NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt

Apache License 2.0

533 stars 135 forks source link

What are topk and keeptopk parameters? #26

Closed DJT777 closed 1 year ago

DJT777 commented 2 years ago

Can someone please describe to me what the "topk" and "keeptopk" parameters are? They are found in the conversion to onnx, in the arguments given for running the python script.

mchi-zg commented 1 year ago

Please refer to TensorRT - batchedNMSPlugin : https://github.com/NVIDIA/TensorRT/tree/main/plugin/batchedNMSPlugin