ppogg/YOLOv5-Lite

请教大佬个问题,推理文件夹时,崩溃。engine->getNbBindings()的值等于4了,如何解决?

retioa11 opened this issue · 0 comments

image

修改的代码:
image

pt导出onnx参数设置:
image

onnx导出trt的命令:
F:/ThirdPartyLibrary/TensorRT-8.6.1.6.Windows10.x86_64.cuda-11.8/bin/trtexec.exe --onnx=E:/33EdgeAccelerate/CodeGitHubDownload0122/YOLOv5-Lite-1.5/runs/train/exp12/weights/best.batch1.onnx --saveEngine=E:/33EdgeAccelerate/CodeGitHubDownload0122/YOLOv5-Lite-1.5/runs/train/exp12/weights/best.batch1.onnx.engine --fp16 --workspace=1000