IBuilder* builder = createInferBuilder(gLogger);
INetworkDefinition* network = builder->createNetwork();
for (int i = 0; i < network->getNbInputs(); ++i)
{
std::string tName = network->getInput(i)->getName();
std::cout << tName << " \n";
}
// set dynamic range for layer output tensors
for (int i = 0; i < network->getNbLayers(); ++i)
{
auto lyr = network->getLayer(i);
for (int j = 0, e = lyr->getNbOutputs(); j < e; ++j)
{
std::string tName = lyr->getOutput(j)->getName();
std::cout << tName << " \n";
}
}
TensorRT解析caffemodel,获取每一层的名字
最新推荐文章于 2023-03-07 21:46:37 发布