RapidAI / RapidStructure

版面分析 | 表格识别 | 文档方向分类
Apache License 2.0
182 stars 14 forks source link

转ONNX问题 #12

Closed weilx2267 closed 9 months ago

weilx2267 commented 9 months ago

您好,打扰了,我这边按照paddlestructure指定的方式训练版面分析,获取到了推断模型,然后使用paddle2onnx进行转onnx,然后导入到本项目下,推理的时候查看了一下输入输出: 输入:['image', 'scale_factor'] 输出:['multiclass_nms3_0.tmp_0', 'multiclass_nms3_0.tmp_2'] 使用您提供的模型文件输入输出是: 输入:['image'] 输出:['transpose_0.tmp_0', 'transpose_2.tmp_0', 'transpose_4.tmp_0', 'transpose_6.tmp_0', 'transpose_1.tmp_0', 'transpose_3.tmp_0', 'transpose_5.tmp_0', 'transpose_7.tmp_0'] 请问转onnx过程有其它的哪些操作,才能完全适配您这个项目?