search

PaddlePaddle / Paddle2ONNX

ONNX Model Exporter for PaddlePaddle
Apache License 2.0
731 stars 172 forks source link

Paddle2ONNX 转换成功后支持将模型转为 TRT 格式吗？ #1075

Open Shrinco opened 1 year ago

Shrinco commented 1 year ago

问题描述 使用paddle2onnx 将maskrcnn的模型转为onnx后，利用tensorrt将onnx格式的模型转位trt时出现了下列错误

[04/04/2023-15:48:09] [E] [TRT] ModelImporter.cpp:773: While parsing node number 498 [Reshape -> "Reshape_0"]: [04/04/2023-15:48:09] [E] [TRT] ModelImporter.cpp:774: --- Begin node --- [04/04/2023-15:48:09] [E] [TRT] ModelImporter.cpp:775: input: "conv2d_53.b_0" input: "Constant_0" output: "Reshape_0" name: "Reshape_0" op_type: "Reshape" [04/04/2023-15:48:09] [E] [TRT] ModelImporter.cpp:776: --- End node --- [04/04/2023-15:48:09] [E] [TRT] ModelImporter.cpp:779: ERROR: ModelImporter.cpp:163 In function parseGraph: [6] Invalid Node - Reshape_0 Attribute not found: allowzero [04/04/2023-15:48:09] [E] Failed to parse onnx file [04/04/2023-15:48:09] [I] Finish parsing network model [04/04/2023-15:48:09] [E] Parsing model failed [04/04/2023-15:48:09] [E] Failed to create engine from model or file. [04/04/2023-15:48:09] [E] Engine set up failed

报错截图 image

yeliang2258 commented 1 year ago

请问是什么模型，可以提供一下模型，我这边来看看

Shrinco commented 1 year ago

请问是什么模型，可以提供一下模型，我这边来看看

您好，模型是 mask_rcnn_r50_fpn_1x_coco 和PaddleDetection里官方提供的一致，没有对网络做过更改。

yeliang2258 commented 1 year ago

辛苦提供转换时候的Paddle模型，我来看看

Shrinco commented 1 year ago

您好，我把模型放在下面的连接了。您可以用百度云下载它。 链接：https://pan.baidu.com/s/1panWdC_xYNmm9PSCloxfvQ?pwd=8bv8 提取码：8bv8

yeliang2258 commented 1 year ago

这个模型里面有distribute_fpn_proposals, generate_proposals_v2两个OP不支持导出为ONNX，试试用PaddleInference-TRT部署吧