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 316 forks source link

TRT8分支,yolov4-tiny bug、不支持yolov3-tiny #145

Closed richjjj closed 2 years ago

richjjj commented 2 years ago

TRT8分支下,yolov4-tiny的官方模型能正常预测,但是自己训练模型预测结果为空(同一模型在tensort7下能正常识别);yolov3-tiny转成trt engine会报错。

enazoe commented 2 years ago

@chongzhong trt8的代码现在不支持yolov3-tiny和yolov4-tiny

richjjj commented 2 years ago

@enazoe 谢啦