Paddle2ONNX icon indicating copy to clipboard operation
Paddle2ONNX copied to clipboard

paddle.onnx.export时output的name怎么修改

Open Agino-ltp opened this issue 4 years ago • 3 comments

image

image

input的name可以这样设置为“image”,那output的name怎么设置呢?怎么把save_infer_model/scale_0.tmp_0修改成想要设置的name呢? x_spec = paddle.static.InputSpec(shape=[1, 3, 48, 192], dtype='float32', name="image")

pytorch可以通过output_names设置,paddle的改怎么去设置呢 torch.onnx.export(model, args, f, export_params=True, verbose=False, training=False, input_names=None, output_names=None)

Agino-ltp avatar Feb 18 '22 12:02 Agino-ltp

参考这个方式进行修改 https://github.com/jiangjiajun/PaddleUtils/tree/main/onnx#3-%E4%BF%AE%E6%94%B9%E6%A8%A1%E5%9E%8B%E4%B8%AD%E9%97%B4%E8%8A%82%E7%82%B9%E5%91%BD%E5%90%8D%E5%8C%85%E5%90%AB%E8%BE%93%E5%85%A5%E8%BE%93%E5%87%BA%E9%87%8D%E5%91%BD%E5%90%8D

jiangjiajun avatar Feb 21 '22 01:02 jiangjiajun

参考这个方式进行修改 https://github.com/jiangjiajun/PaddleUtils/tree/main/onnx#3-%E4%BF%AE%E6%94%B9%E6%A8%A1%E5%9E%8B%E4%B8%AD%E9%97%B4%E8%8A%82%E7%82%B9%E5%91%BD%E5%90%8D%E5%8C%85%E5%90%AB%E8%BE%93%E5%85%A5%E8%BE%93%E5%87%BA%E9%87%8D%E5%91%BD%E5%90%8D

可以修改,谢谢

Agino-ltp avatar Feb 21 '22 04:02 Agino-ltp

另外我们也准备了一份问卷,方便的话请抽出半分钟时间帮完成问卷 https://iwenjuan.baidu.com/?code=r8hu2s 帮助Paddle2ONNX做得更好

jiangjiajun avatar Feb 22 '22 02:02 jiangjiajun