Hello @linghu8812,
Thank you for you great effort. I have a yolov4 model trained on custom dataset with 5 classes. Is it possible to convert it to tensorRT.
I have changed the number of classes in export_onnx.py for 80 to 5 and loaded my weigths and cfg files.
Hello @linghu8812, Thank you for you great effort. I have a yolov4 model trained on custom dataset with 5 classes. Is it possible to convert it to tensorRT.
I have changed the number of classes in
export_onnx.py
for 80 to 5 and loaded my weigths and cfg files.After conversion I get the following error:
Is there anything else I have to do? Thanks
note: I have tested the pretrained model and works fine but my custom model gives me the above error.