spacewalk01

Thanks, @jstumpin I consider your suggestion!

@jstumpin I am using RTX-3070 gpu with 8Gb memory for running Yolov4 model on TensorRT with fp16 precision. I've obtained 135fps on average with the preprocessing kernel function implemented by...

source code please

@xenova's solution works

Got same error in detect.py

It works!

So, the correct one is as follows, right? ``` line 903 in utils/general.py prediction = prediction[0][1] ```

any progress on tensorrt implementation? thanks

Nice project, please provide both encoder and decoder onnx files

> in file `onnx_add_nms_plugin.py` I changed line 41 from `op="BatchedNMS_TRT",` to `op="BatchedNMSDynamic_TRT",`. Using a modified `onnx_add_nms_plugin.py` I generated a new onnx model and it worked successfully. It works for me...