search

PaddlePaddle / Paddle2ONNX

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

请问Paddle2ONNX可以配合哪一个PaddleDetection模型使用？ #1423

Open 188080501 opened 2 hours ago

188080501 commented 2 hours ago

我尝试了 PP-YOLOE_plus-L和YOLOX-L模型，可以正常导出ONNX，但是加载运行时报错 Op (Gather) [ShapeInferenceError] data tensor must have rank >= 1"

尝试多个方法都无法解决这个问题，也有其他人提了同个issuse，如下： https://github.com/PaddlePaddle/Paddle2ONNX/issues/1375

然后我尝试了FasterRCNN-ResNet101模型，但是转换时报错，无法正常得到onnx文件 LodTensorArray is not supported.

Paddle2ONNX已经使用了最新版本1.2.11 PaddleX训练的模型，使用的是3.0-beta1

所以请教一下，如果我想ONNX做目标检测，我应该使用哪个模型？或者说是我参数错误导致？

188080501 commented 1 hour ago

语义分割 Deeplabv3_Plus-R50 模型 实例分割 Mask-RT-DETR-X 模型 我都已经跑通，得到ONNX模型并且正常推理