mge 转onnx 报错

#1

conert mge_to_onnx -i crestereo_eth3d.mge -o out.onnx 报错。
MegEngine 1.11.0
megfile 1.0.2
mgeconvert 1.0.2
模型为你们官方的双目匹配模型 crestereo_eth3d.mge
报错信息

image
image1432×524 81.3 KB

#2

请问能麻烦你准确地告诉我原模型在哪个地方嘛?

#3

原因：首先，这个问题是因为你下载的文件其实是一个模型的权重字典，模型结构是由该项目里面的代码来指定的，而 mgeconvert 转换模型是需要模型结构的。所以出现了你遇到的问题。
模型加载详情参考(https://github.com/megvii-research/CREStereo/blob/ad3a1613bdedd88b93247e5f002cb7c80799762d/test.py#L17)

解决办法:
dump出具有模型结构的序列化文件，并通过mgeconvert 转换到你所需要平台的模型。

  1. 首先，你可以 https://megengine.megvii-inc.com/user-guide/deployment/ 参考一下,模型的部署路线。
  2. 然后你可以在 https://megengine.megvii-inc.com/user-guide/model-development/jit/dump.html# 这里学习如何 dump 出一个具有模型结构的mge文件,或者在https://megengine.megvii-inc.com/user-guide/model-development/traced_module/quick-start.html 学习如果dump 出一个tm模型。
  3. 然后通过 mgeconvert 的命令行接口或者编程接口来转换模型。

希望对你有所帮助。

#4

算子不支持

image
image1011×398 51.3 KB