[QUESTIONS] .wts file and plugin file

[devaale]

[QUESTIONS] .wts file and plugin file

[philipp-schmidt]

You don't need the .wts to compile the layer plugin, but you need to load some weights if you want to create the yolov4 tensorrt engine. You can use the .wts file that we provide in Google Drive for the default yolov4 weights which you would get if you dowload them from darknet (the repo yolov4 comes from and was implemented in). Or you can use your own .wts file from your own training. If your training involved changing any of the network parameters you might have to change the values in the code as well, yes. So especially the lines that you highlighted and maybe also the yolo layer anchors if you changed them.

[philipp-schmidt]

If you post more details (e.g. your network config) we might be able to help you.

[devaale]

@philipp-schmidt Sorry for late response. Would appreciate the help with figuring this out. Link to my .cfg file: https://drive.google.com/drive/folders/1A65yRs2Lbw7n6LYeC_mps9ZMO9gRuVlK?usp=sharing