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.18k stars 313 forks source link

是不是不支持yolov5的2.0以前的版本? #56

Open gxingjie20 opened 3 years ago

gxingjie20 commented 3 years ago

是不是不支持yolov5的2.0以前的版本?我用的刚开始的版本,可以生成cfg和weight,但是在detector_init(config_v5)会报错

enazoe commented 3 years ago

@gxingjie20 支持2.0再以前的我不确定,2.0需要改动一块地方

enazoe commented 3 years ago

@gxingjie20 当然改动不大的话我可以帮你改一下,因为内部的实现是框架性的,等我看下v1.0的结构,你可以加群讨论,群号在readme界面

gxingjie20 commented 3 years ago

@enazoe 好的 多谢