your project is helpful! thanks a lot!
however, I use TensorRT >= 7.1 with CUDA 10.2, and your project could not run, because the version not match.
could you provide a docker image for your project, it will help a lot!
I'd like to know about the difference between your jde model and the official model in which the 3 * (5 + classes_num) + embedding dimensions feature directly from the last same convolution simultaneously.