Giter Site home page Giter Site logo

Comments (18)

novioleo avatar novioleo commented on April 26, 2024

具体一点。
另外不建议转mnn。

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

具体一点。
另外不建议转mnn。

你好,我文字识别crnn模型里面有lstm部分,转onnx没问题,onnx转mnn的时候出现如下:

[06:42:53] :31: ONNX Model ir version: 6
[06:42:53] :155: Check it out ==> 726 has empty input, the index is 4
[06:42:53] :155: Check it out ==> 777 has empty input, the index is 4
Start to Optimize the MNN Net...
[06:42:54] :64: Inputs: input
[06:42:54] :69: Outputs: output

mnn模型输出结果是不对的。
onnx模型输出维度是(1,x,2671),x是不固定的,
但是mnn模型输出固定式(2,1,96)

请问mnn支持lstm转换吗?怎么解决?

onnx中的lstm部分结构图如下:
image

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

具体一点。
另外不建议转mnn。

我们走的都是onnx -> mnn这条路,所以 。。。。必须要转mnn。。。。。。。。。。。。。。。。。。

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

image

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

@cendelian 你是用什么工具转的?

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

@cendelian 你是用什么工具转的?

我先转成onnx,onnx推理是OK的,与pytorch一致。
然后用mnn转换工具将onnx转成mnn,mnn转换指令如下:
mnnconvert -f ONNX --modelFile xxx.onnx --MNNModel xxx.mnn

转换会出现

[06:42:53] :155: Check it out ==> 726 has empty input, the index is 4
[06:42:53] :155: Check it out ==> 777 has empty input, the index is 4

貌似lstm转换不对还是不支持啥的??
并且转换后的mnn模型输出形状固定是(2, 1, 96)。。。。。。
正确的输出形状应该是(1,x, 字符数长度)。。

大佬麻烦帮忙看下哈,这个几天了。。。没搞定。。

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

@cendelian https://convertmodel.com/ 先试试这个

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

@cendelian https://convertmodel.com/ 先试试这个

这个网址联转mnn都转不成功了。。报错如下:
image

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

@cendelian
你把模型分别导出为多个部分,例如backbone 、neck、head多个部分。每个onnx你进行转换下。这样问题也好排查

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

@cendelian
你把模型分别导出为多个部分,例如backbone 、neck、head多个部分。每个onnx你进行转换下。这样问题也好排查

怎么导出。。。

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

每个module export到onnx啊。backbone、neck、head都是一个network module嘛。

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

每个module export到onnx啊。backbone、neck、head都是一个network module嘛。

要加载权重参数后才能导出吧,可是加载后就是一个整体了,,怎么划分开导出

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

你先加载参数嘛,加载完了之后这个net里面有多个module,torch.onnx.export(net.backbone) 这个样子的方式导出其中的某个module
@cendelian

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

@cendelian 如何了?

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

@cendelian 如何了?

没解决。。。。。。

from pytorchocr.

novioleo avatar novioleo commented on April 26, 2024

@cendelian 如何了?

没解决。。。。。。

@cendelian
是没去弄还是遇到问题了?

from pytorchocr.

dengfenglai321 avatar dengfenglai321 commented on April 26, 2024

已经解决, 发现onnx转mnn有empty提示, 但是不影响MNN的推理结果。。。。。。。。。。。。。。。。。。。。。
所以这个提示可以忽略不计

from pytorchocr.

nissansz avatar nissansz commented on April 26, 2024

具体一点。
另外不建议转mnn。

你好,我文字识别crnn模型里面有lstm部分,转onnx没问题,onnx转mnn的时候出现如下:

[06:42:53] :31: ONNX Model ir version: 6 [06:42:53] :155: Check it out ==> 726 has empty input, the index is 4 [06:42:53] :155: Check it out ==> 777 has empty input, the index is 4 Start to Optimize the MNN Net... [06:42:54] :64: Inputs: input [06:42:54] :69: Outputs: output

mnn模型输出结果是不对的。 onnx模型输出维度是(1,x,2671),x是不固定的, 但是mnn模型输出固定式(2,1,96)

请问mnn支持lstm转换吗?怎么解决?

onnx中的lstm部分结构图如下: image

你好,可以发一下pth转onnx的代码吗?

from pytorchocr.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.