报错信息如下:
博主在进行部署时遇到的错误,查阅以后发现是一个很简单的问题,解决方法如下:
在序列化模型之前,需要给模型的输入输出设定名字
即在buildserializeengine之前添加
network->getInput(0)->setName("input");
network->getOutput(0)->setName("output");
名字随便起,只要保证在反序列化时名字相同即可
auto inputIndex = engine->getBindingIndex("input");
auto outputIndex = engine->getBindingIndex("output");