Closed devaale closed 2 years ago
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.
If you post more details (e.g. your network config) we might be able to help you.
@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
I was wondering do I need to use my own generated .wts file (from .weights file) when generating plugins lib. ? (
liblayerplugin,so
) This step here:.wts
from google drive or do I need to convert my own.weights
/.cfg
files to.wts
? - Will it affect results ?Additionally wanted to ask: when generating
liblayerplugin.so
should I use my yolov4.cfg file and edit code parts to match my data ? For examplenetworks/yolov4.h
file: