xunxuntu

我用trtexec工具生成了yolov5s.trt，并放在了weights下，然后执行 ./object_detection yolov5 /home/tuc/yolov5-6.1/data/images 命令，显示了推理时间，却没有在保存的图片上将检测框画出来。 如下图  

是的，是最新的代码。昨天才git下来的

我把dynamic设为了0，还是不起作用。 您是怎么得到trt文件的，我是用这个命令得到的： ```shell root@rtx2060:/usr/local/TensorRT-8.4.3.1/bin# ./trtexec --onnx=/home/tuc/yolov5-6.1/weights/yolov5s.onnx --saveEngine=/home/tuc/yolov5-6.1/weights/outfpbest.trt --workspace=2048 --best # 导出onnx是用这个命令，yolov5版本是6.1 python export.py --weights=yolov5s.pt --dynamic --simplify --include=onnx --opset 11 ``` > 会不会和环境有关 > 我用的是cuda 11.2 , cudnn 8.4.0 , TensorRT-8.4.3.1...

> 那就已经是float了，我刚刚测试了这两张图应该是没有问题的，因此我认为还是trt转换出了问题，你可以把config中的dynamic设置为0重新生成一下（因为从报错信息来看onnx为静态文件） 之前的报错信息在我导出 trt 模型后，放在weights路径下就自动消失了。

很奇怪，我删除了原先的onnx 和 trt 文件 重新使用export.py导出了onnx，`python export.py --weights=weights/yolov5s.pt --include=onnx` ``` (tc_yolov5) root@rtx2060:/home/tuc/yolov5-6.1# python export.py --weights=weights/yolov5s.pt --include=onnx export: data=data/coco128.yaml, weights=['weights/yolov5s.pt'], imgsz=[640, 640], batch_size=1, device=cpu, half=False, inplace=False, train=False, optimize=False, int8=False, dynamic=False, simplify=False, opset=12,...

会不会 是你代码里的后处理的问题呀，能够打印推理的时间，但是不能画框，是不是检测的那部分。

> 我暂时也无法定位问题，因为我复现不出来。。 博主，您好，我将我本地的模型和其代码上传到 https://github.com/xunxuntu/check_todo.git 这里了，您有空可以git 下来帮我看看吗 😁😁😁😁😁😁😁😁😁