enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.19k stars 315 forks source link

Other yolo models #166

Open ttdd11 opened 2 years ago

ttdd11 commented 2 years ago

Are the scaled models (ie. yolov4-p5.cfg) supported?

Also, in the .cfg for the yolov4-csp has a "logistics" activation, is it okay if that is changed to linear in the config (directly from the yolo github).

Incredible work, thank you!