Question

我正在尝试使用带有TensorRT的Faster R-CNN的预训练Caffe模型。当我尝试使用此行构建引擎

engine = trt.utils.caffe_to_trt_engine(G_LOGGER,
                                   MODEL_PROTOTXT,
                                   CAFFE_MODEL,
                                   1,
                                   1 << 20,
                                   OUTPUT_LAYERS,
                                   trt.infer.DataType.FLOAT)

我收到以下错误：

could not parse layer type IPlugin
[TensorRT] ERROR: Failed to parse caffe model
  File "/usr/lib/python2.7/dist-packages/tensorrt/utils/_utils.py", line 352, in caffe_to_trt_engine
    assert(blob_name_to_tensor)
Traceback (most recent call last):
  File "tensorRT_conversionFasterRCNN.py", line 38, in <module>
    trt.infer.DataType.FLOAT)
  File "/usr/lib/python2.7/dist-packages/tensorrt/utils/_utils.py", line 360, in caffe_to_trt_engine
    raise AssertionError('Caffe parsing failed on line {} in statement {}'.format(line, text))
AssertionError: Caffe parsing failed on line 352 in statement assert(blob_name_to_tensor)

我正在使用TensorRT-4.0.1.6提供的“ faster_rcnn_test_iplugin.prototxt”示例文件，其层应该与TensorRT兼容。

有人知道我是否缺少什么吗？谢谢

尝试使用TensorRT构建更快的R-CNN引擎时，“无法解析层类型IPlugin [TensorRT]错误：无法解析caffe模型”

0 个答案: