search

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

使用自己训练yolov5l模型，生成engine后检测锚框不准，在tensorrtx工程上可以正常运行，请问有可能是那部分的问题 #176

Closed liuyang9709 closed 2 years ago

liuyang9709 commented 2 years ago

使用自己训练yolov5l v6.0模型，生成engine后检测锚框不准，在tensorrtx工程上可以正常运行，请问有可能是那部分的问题 使用官方模型可以正常运行 ubuntu20, CUDA11.1, TensorRT 8.0.2.3 GPU: NVIDIA rtx a4000

liuyang9709 commented 2 years ago

找到问题了，项目中锚框部分直接使用的是配置文件中的。