This supplies TRT model (with batchedNMSPlugin) for yolov7 model.
- onnx: 1.8.0
- torch: 1.9.0a0+df837d0
- onnx-graphsurgeon: 0.2.8
- tensorrt: 7.2.2.3
- CUDA: 11.2
- Driver Version: 460.73.01
python export.py --weights yolov7.pt
python add_nmsplugin.py --onnx_path yolov7.onnx
trtexec --onnx=yolov7_gs.onnx --fp16 --workspace --saveEngine=yolov7_gs.trt
python detect_trt_plugin.py --model yolov7_gs.trt --image_folder inference/images --classes coco.yaml