onnx_tensorrt_project icon indicating copy to clipboard operation
onnx_tensorrt_project copied to clipboard

Published 20 hours ago • verified publisher icon ttanzhiqiang

yolov5_detector.cpp在多批次下报错

Open Tristesse-stk opened this issue 3 years ago • 1 comments

如题，yolov5_detector.cpp在推理阶段时如果输入的batch是多个的话（即大于1） yolov5-detector输入多batch 那么会报以下错误： yolov5-detector报错 请问有可能是什么原因导致的呢？

Tristesse-stk avatar Sep 24 '21 05:09 Tristesse-stk

你需要重新生成onnx（batch=2），然后再使用这个代码生成trt文件

ttanzhiqiang avatar Sep 24 '21 05:09 ttanzhiqiang