Assertion'engine->getNbBindings()==2'failed

linghu8812 / tensorrt_inference

708 stars 208 forks source link

Closed Egorundel closed 1 year ago

Egorundel commented 1 year ago

嗨！在你的模型导出代码中没有NMS模块，我肯定需要使用BatchedNMS_TRT，但是，正因为如此，我得到错误"Assertion'engine->getNbBindings()==2'failed。"

我已经推导出了我的getNbBindings的数量，我有5个。我可以在不影响代码的情况下将assert(engine->getNbBindings()==2)更改为assert(engine->getNbBindings()==5)吗？

Egorundel commented 1 year ago

这个问题是通过更换来解决的 void *buffers[2]; 在模型中。h上 void *buffers[5];

并在模型中。cpp改线了 assert(engine->getNbBindings()==2); 到 assert(engine->getNbBindings()==5);

如果您在模型中使用NMS插件，请执行此操作:)