3

我试图使用 TensorRT Python API 来转换模型。定义连接层时出错,代码为:

concatLayer = network.add_concatenation([conv1.get_output(0),conv2.get_output(0)],2)

错误信息是:

TypeError: in method 'NetworkDefinition_add_concatenation', argument 2 of type 'nvinfer1::ITensor *const *'

' conv1.get_output(0)' 将返回一个ITensor*变量,我应该如何将其转换为的类型ITensor *const*

似乎 TensorRT python API 是用 SWIG 从其 C++ 版本包装的,add_concatenation() 的 API 参考是:

addConcatenation(ITensor *const *inputs, int nbInputs)=0 -> IConcatenationLayer *

向网络参数添加连接层:

inputs (*) – the input tensors to the layer
nbInputs (*) – the number of input tensors
also (See) –

回报:

the new concatenation layer, or null if it could not be created
**Warning** (All tensors must have the same dimensions for all dimensions except for channel.)
4

0 回答 0