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,获取每一层的名字
最新推荐文章于 2024-07-19 11:43:03 发布