Support for different number of classes and yolov4-px models

jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet

https://jkjung-avt.github.io/

MIT License

1.75k stars 547 forks source link

Support for different number of classes and yolov4-px models #497

Closed lpkoh closed 2 years ago

lpkoh commented 3 years ago

Hi,

Can I check if support for "custom models" means that you can convert darknet yolo models to onnx that differ not only in image dimension (e.g. 640x 640 -> 640x320), but also different number of classes? Thanks!

jkjung-avt commented 3 years ago

Yes.

lpkoh commented 3 years ago

Omg thank you man