PaddlePaddle / Paddle2ONNX

ONNX Model Exporter for PaddlePaddle
Apache License 2.0
672 stars 156 forks source link

多batch推理 #1051

Open bixiaopeng0 opened 1 year ago

bixiaopeng0 commented 1 year ago

请将下面信息填写完整,便于我们快速解决问题,谢谢!

问题描述 请在此处详细的描述报错信息

ppyoloe支持onnx batch>1推理么,如果不支持,具体是为什么呢? 更多信息 :

报错截图

其他信息

yeliang2258 commented 1 year ago

由于multiclass_nms3 op的原因,目前只支持batch size为1

bixiaopeng0 commented 1 year ago

如果去掉后处理,是支持的哈

yeliang2258 commented 1 year ago

是的,去掉后处理,将nms放到外面做,就可以支持多batch